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