refactor/pretrainedconfig_device_amps_args
#11
by
imstevenpmwork
HF Staff
- opened
- 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",
|