File size: 461 Bytes
dcf7d64 8f3b4d8 dcf7d64 8f3b4d8 dcf7d64 8f3b4d8 dcf7d64 8f3b4d8 dcf7d64 |
1 2 3 4 5 6 7 8 9 10 11 12 |
This branch contains the model weights obtained from training on the original Diffusion Policy repository.
This is the command that was used for training:
```bash
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`](convert_weights.py).
|