fix/add_args_to_nested_policy_config
Browse files- train_config.json +2 -0
train_config.json
CHANGED
|
@@ -104,6 +104,8 @@
|
|
| 104 |
"policy": {
|
| 105 |
"type": "diffusion",
|
| 106 |
"n_obs_steps": 2,
|
|
|
|
|
|
|
| 107 |
"normalization_mapping": {
|
| 108 |
"VISUAL": "MEAN_STD",
|
| 109 |
"STATE": "MIN_MAX",
|
|
|
|
| 104 |
"policy": {
|
| 105 |
"type": "diffusion",
|
| 106 |
"n_obs_steps": 2,
|
| 107 |
+
"device": "cuda",
|
| 108 |
+
"use_amp": false,
|
| 109 |
"normalization_mapping": {
|
| 110 |
"VISUAL": "MEAN_STD",
|
| 111 |
"STATE": "MIN_MAX",
|