Upload folder using huggingface_hub
Browse files- config.json +2 -0
- train_config.json +0 -2
config.json
CHANGED
|
@@ -38,6 +38,8 @@
|
|
| 38 |
"n_action_steps": 8,
|
| 39 |
"n_groups": 8,
|
| 40 |
"n_obs_steps": 2,
|
|
|
|
|
|
|
| 41 |
"noise_scheduler_type": "DDPM",
|
| 42 |
"normalization_mapping": {
|
| 43 |
"ACTION": "MIN_MAX",
|
|
|
|
| 38 |
"n_action_steps": 8,
|
| 39 |
"n_groups": 8,
|
| 40 |
"n_obs_steps": 2,
|
| 41 |
+
"device": "cuda",
|
| 42 |
+
"use_amp": false,
|
| 43 |
"noise_scheduler_type": "DDPM",
|
| 44 |
"normalization_mapping": {
|
| 45 |
"ACTION": "MIN_MAX",
|
train_config.json
CHANGED
|
@@ -178,8 +178,6 @@
|
|
| 178 |
"output_dir": "outputs/train/2025-01-17/11-51-15_pusht_diffusion",
|
| 179 |
"job_name": "pusht_diffusion",
|
| 180 |
"resume": false,
|
| 181 |
-
"device": "cuda",
|
| 182 |
-
"use_amp": false,
|
| 183 |
"seed": 100000,
|
| 184 |
"num_workers": 4,
|
| 185 |
"batch_size": 64,
|
|
|
|
| 178 |
"output_dir": "outputs/train/2025-01-17/11-51-15_pusht_diffusion",
|
| 179 |
"job_name": "pusht_diffusion",
|
| 180 |
"resume": false,
|
|
|
|
|
|
|
| 181 |
"seed": 100000,
|
| 182 |
"num_workers": 4,
|
| 183 |
"batch_size": 64,
|