Views
No views yet
1lerobot-train \
2 --dataset.repo_id=${HF_USER}/<dataset> \
3 --reward_model.type=robometer \
4 --output_dir=outputs/train/<desired_reward_model_repo_id> \
5 --job_name=lerobot_reward_training \
6 --reward_model.device=cuda \
7 --reward_model.repo_id=${HF_USER}/<desired_reward_model_repo_id> \
8 --wandb.enable=trueoutputs/train/<desired_reward_model_repo_id>/checkpoints/.1from lerobot.rewards import make_reward_model
2
3reward_model = make_reward_model(pretrained_path="<hf_user>/<reward_model_repo_id>")
4reward = reward_model.compute_reward(batch)