diffusion_pusht / README.md
alexandersoare's picture
Upload folder using huggingface_hub
dcf7d64 verified
|
raw
history blame
461 Bytes

This branch contains the model weights obtained from training on the original Diffusion Policy repository.

This is the command that was used for training:

python train.py --config-dir=. --config-name=image_pusht_diffusion_policy_cnn.yaml training.seed=42 logging.name=benchmark

The configuration file image_pusht_diffusion_policy_cnn.yaml is included in this branch.

The weights were converted with convert_weights.py.