Upload folder using huggingface_hub
Browse files- config.json +1 -0
- train_config.json +1 -1
config.json
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
{
|
|
|
|
| 2 |
"n_obs_steps": 1,
|
| 3 |
"normalization_mapping": {
|
| 4 |
"VISUAL": "MEAN_STD",
|
|
|
|
| 1 |
{
|
| 2 |
+
"type": "act",
|
| 3 |
"n_obs_steps": 1,
|
| 4 |
"normalization_mapping": {
|
| 5 |
"VISUAL": "MEAN_STD",
|
train_config.json
CHANGED
|
@@ -105,7 +105,7 @@
|
|
| 105 |
]
|
| 106 |
}
|
| 107 |
},
|
| 108 |
-
"device": "
|
| 109 |
"use_amp": false,
|
| 110 |
"chunk_size": 100,
|
| 111 |
"n_action_steps": 100,
|
|
|
|
| 105 |
]
|
| 106 |
}
|
| 107 |
},
|
| 108 |
+
"device": "mps",
|
| 109 |
"use_amp": false,
|
| 110 |
"chunk_size": 100,
|
| 111 |
"n_action_steps": 100,
|