refactor/pretrainedconfig_device_amps_args
#3
by
imstevenpmwork
HF Staff
- opened
- config.json +2 -0
- train_config.json +2 -2
config.json
CHANGED
|
@@ -1,6 +1,8 @@
|
|
| 1 |
{
|
| 2 |
"type": "vqbet",
|
| 3 |
"n_obs_steps": 5,
|
|
|
|
|
|
|
| 4 |
"normalization_mapping": {
|
| 5 |
"VISUAL": "IDENTITY",
|
| 6 |
"STATE": "MIN_MAX",
|
|
|
|
| 1 |
{
|
| 2 |
"type": "vqbet",
|
| 3 |
"n_obs_steps": 5,
|
| 4 |
+
"device": "cuda",
|
| 5 |
+
"use_amp": false,
|
| 6 |
"normalization_mapping": {
|
| 7 |
"VISUAL": "IDENTITY",
|
| 8 |
"STATE": "MIN_MAX",
|
train_config.json
CHANGED
|
@@ -104,6 +104,8 @@
|
|
| 104 |
"policy": {
|
| 105 |
"type": "vqbet",
|
| 106 |
"n_obs_steps": 5,
|
|
|
|
|
|
|
| 107 |
"normalization_mapping": {
|
| 108 |
"VISUAL": "IDENTITY",
|
| 109 |
"STATE": "MIN_MAX",
|
|
@@ -175,8 +177,6 @@
|
|
| 175 |
"output_dir": "outputs/train/2025-01-25/20-30-40_pusht_vqbet",
|
| 176 |
"job_name": "pusht_vqbet",
|
| 177 |
"resume": false,
|
| 178 |
-
"device": "cuda",
|
| 179 |
-
"use_amp": false,
|
| 180 |
"seed": 100000,
|
| 181 |
"num_workers": 4,
|
| 182 |
"batch_size": 64,
|
|
|
|
| 104 |
"policy": {
|
| 105 |
"type": "vqbet",
|
| 106 |
"n_obs_steps": 5,
|
| 107 |
+
"device": "cuda",
|
| 108 |
+
"use_amp": false,
|
| 109 |
"normalization_mapping": {
|
| 110 |
"VISUAL": "IDENTITY",
|
| 111 |
"STATE": "MIN_MAX",
|
|
|
|
| 177 |
"output_dir": "outputs/train/2025-01-25/20-30-40_pusht_vqbet",
|
| 178 |
"job_name": "pusht_vqbet",
|
| 179 |
"resume": false,
|
|
|
|
|
|
|
| 180 |
"seed": 100000,
|
| 181 |
"num_workers": 4,
|
| 182 |
"batch_size": 64,
|