refactor/pretrainedconfig_device_amps_args (#11)
Browse files- Upload folder using huggingface_hub (4956cd821f0ae43543f172ff3e9a474e88d4f801)
Co-authored-by: Steven Palma <imstevenpmwork@users.noreply.huggingface.co>
- config.json +2 -0
config.json
CHANGED
|
@@ -1,6 +1,8 @@
|
|
| 1 |
{
|
| 2 |
"type": "pi0",
|
| 3 |
"n_obs_steps": 1,
|
|
|
|
|
|
|
| 4 |
"normalization_mapping": {
|
| 5 |
"VISUAL": "IDENTITY",
|
| 6 |
"STATE": "MEAN_STD",
|
|
|
|
| 1 |
{
|
| 2 |
"type": "pi0",
|
| 3 |
"n_obs_steps": 1,
|
| 4 |
+
"device": "cuda",
|
| 5 |
+
"use_amp": false,
|
| 6 |
"normalization_mapping": {
|
| 7 |
"VISUAL": "IDENTITY",
|
| 8 |
"STATE": "MEAN_STD",
|