Update train_config.json
Browse files- train_config.json +2 -2
train_config.json
CHANGED
|
@@ -105,8 +105,6 @@
|
|
| 105 |
]
|
| 106 |
}
|
| 107 |
},
|
| 108 |
-
"device": "cuda",
|
| 109 |
-
"use_amp": false,
|
| 110 |
"horizon": 16,
|
| 111 |
"n_action_steps": 8,
|
| 112 |
"drop_n_last_frames": 7,
|
|
@@ -149,6 +147,8 @@
|
|
| 149 |
"scheduler_name": "cosine",
|
| 150 |
"scheduler_warmup_steps": 500
|
| 151 |
},
|
|
|
|
|
|
|
| 152 |
"output_dir": "outputs/train/act-pick-put02",
|
| 153 |
"job_name": "act-pick-put01",
|
| 154 |
"resume": false,
|
|
|
|
| 105 |
]
|
| 106 |
}
|
| 107 |
},
|
|
|
|
|
|
|
| 108 |
"horizon": 16,
|
| 109 |
"n_action_steps": 8,
|
| 110 |
"drop_n_last_frames": 7,
|
|
|
|
| 147 |
"scheduler_name": "cosine",
|
| 148 |
"scheduler_warmup_steps": 500
|
| 149 |
},
|
| 150 |
+
"device": "cuda",
|
| 151 |
+
"use_amp": false,
|
| 152 |
"output_dir": "outputs/train/act-pick-put02",
|
| 153 |
"job_name": "act-pick-put01",
|
| 154 |
"resume": false,
|