Upload folder using huggingface_hub
Browse files- config.json +2 -0
config.json
CHANGED
|
@@ -33,6 +33,8 @@
|
|
| 33 |
"n_action_steps": 8,
|
| 34 |
"n_groups": 8,
|
| 35 |
"n_obs_steps": 2,
|
|
|
|
|
|
|
| 36 |
"noise_scheduler_type": "DDIM",
|
| 37 |
"num_inference_steps": 10,
|
| 38 |
"num_train_timesteps": 100,
|
|
|
|
| 33 |
"n_action_steps": 8,
|
| 34 |
"n_groups": 8,
|
| 35 |
"n_obs_steps": 2,
|
| 36 |
+
"device": "cuda",
|
| 37 |
+
"use_amp": false,
|
| 38 |
"noise_scheduler_type": "DDIM",
|
| 39 |
"num_inference_steps": 10,
|
| 40 |
"num_train_timesteps": 100,
|