Amandee
/

diffusion_transformer_pusht

Model card Files Files and versions Community

Amandee commited on Feb 15

Commit

957b9b3

·

verified ·

1 Parent(s): 7a0967a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -21,7 +21,7 @@ The model was trained using [LeRobot's training script](https://github.com/huggi
 python lerobot/scripts/train.py \
   hydra.run.dir=outputs/train/diffusion_pusht \
   hydra.job.name=diffusion_pusht \
-  policy=diffusion_transformer training.save_model=true \
   env=pusht \
   env.task=PushT-v0 \
   dataset_repo_id=lerobot/pusht \

 python lerobot/scripts/train.py \
   hydra.run.dir=outputs/train/diffusion_pusht \
   hydra.job.name=diffusion_pusht \
+  policy=diffusion training.save_model=true \
   env=pusht \
   env.task=PushT-v0 \
   dataset_repo_id=lerobot/pusht \