yor-patch-policy-checkpoints / code /configs /train_pusht_diffusion_1gpu.yaml
szcharlesji's picture
Add files using upload-large-folder tool
94f697a verified
Raw History Blame Contribute Delete
283 Bytes
# Push-T diffusion policy, single-GPU embedding-precompute run.
defaults:
- train_pusht_diffusion
- override /encoder: dino_patch
- override /cluster: local
- _self_
batch_size: 256
epochs: 1200
eval_freq: 10
eval_on_env_freq: 10
precompute_embeddings: true
num_workers: 0