Downloads · 30 days
8
19% of all-time downloads
lilkm/vf_stackblocks_iter1
vf_stackblocks_iter1 is a robotics model from lilkm. Use it for the robotics task on the model card, and read the license before you ship it in a product. It is set up for lerobot. The card lists the license as apache-2.0.
Reward model type not recognized — please update this template.
Downloads · 30 days
8
19% of all-time downloads
All-time downloads
42
Public
Parameters
1.4B
5.6 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors5.6 GB · 100%
From the Hugging Face model README
Reward model type not recognized — please update this template.
This reward model has been trained and pushed to the Hub using LeRobot. See the full documentation at LeRobot Docs.
lerobot-train \
--dataset.repo_id=${HF_USER}/<dataset> \
--reward_model.type=distributional_value_function \
--output_dir=outputs/train/<desired_reward_model_repo_id> \
--job_name=lerobot_reward_training \
--reward_model.device=cuda \
--reward_model.repo_id=${HF_USER}/<desired_reward_model_repo_id> \
--wandb.enable=true
Writes checkpoints to outputs/train/<desired_reward_model_repo_id>/checkpoints/.
from lerobot.rewards import make_reward_model
reward_model = make_reward_model(pretrained_path="<hf_user>/<reward_model_repo_id>")
reward = reward_model.compute_reward(batch)