File size: 278 Bytes
94f697a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | # 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
|