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 |
"chunk_size": 50,
|
| 111 |
"n_action_steps": 50,
|
| 112 |
"max_state_dim": 32,
|
|
@@ -148,6 +146,8 @@
|
|
| 148 |
"min_period": 0.004,
|
| 149 |
"max_period": 4.0
|
| 150 |
},
|
|
|
|
|
|
|
| 151 |
"output_dir": "outputs/train/my_finetuned_smolvla",
|
| 152 |
"job_name": "piper-testing",
|
| 153 |
"resume": false,
|
|
|
|
| 105 |
]
|
| 106 |
}
|
| 107 |
},
|
|
|
|
|
|
|
| 108 |
"chunk_size": 50,
|
| 109 |
"n_action_steps": 50,
|
| 110 |
"max_state_dim": 32,
|
|
|
|
| 146 |
"min_period": 0.004,
|
| 147 |
"max_period": 4.0
|
| 148 |
},
|
| 149 |
+
"device": "cuda",
|
| 150 |
+
"use_amp": false,
|
| 151 |
"output_dir": "outputs/train/my_finetuned_smolvla",
|
| 152 |
"job_name": "piper-testing",
|
| 153 |
"resume": false,
|