Upload folder using huggingface_hub
Browse files- config.json +2 -0
- train_config.json +2 -2
config.json
CHANGED
@@ -1,6 +1,8 @@
|
|
1 |
{
|
2 |
"type": "act",
|
3 |
"n_obs_steps": 1,
|
|
|
|
|
4 |
"normalization_mapping": {
|
5 |
"VISUAL": "MEAN_STD",
|
6 |
"STATE": "MEAN_STD",
|
|
|
1 |
{
|
2 |
"type": "act",
|
3 |
"n_obs_steps": 1,
|
4 |
+
"device": "cuda",
|
5 |
+
"use_amp": false,
|
6 |
"normalization_mapping": {
|
7 |
"VISUAL": "MEAN_STD",
|
8 |
"STATE": "MEAN_STD",
|
train_config.json
CHANGED
@@ -102,6 +102,8 @@
|
|
102 |
"policy": {
|
103 |
"type": "act",
|
104 |
"n_obs_steps": 1,
|
|
|
|
|
105 |
"normalization_mapping": {
|
106 |
"VISUAL": "MEAN_STD",
|
107 |
"STATE": "MEAN_STD",
|
@@ -156,8 +158,6 @@
|
|
156 |
"output_dir": "outputs/train/2025-01-25/17-37-25_aloha_act",
|
157 |
"job_name": "aloha_act",
|
158 |
"resume": false,
|
159 |
-
"device": "cuda",
|
160 |
-
"use_amp": false,
|
161 |
"seed": 1000,
|
162 |
"num_workers": 4,
|
163 |
"batch_size": 8,
|
|
|
102 |
"policy": {
|
103 |
"type": "act",
|
104 |
"n_obs_steps": 1,
|
105 |
+
"device": "cuda",
|
106 |
+
"use_amp": false,
|
107 |
"normalization_mapping": {
|
108 |
"VISUAL": "MEAN_STD",
|
109 |
"STATE": "MEAN_STD",
|
|
|
158 |
"output_dir": "outputs/train/2025-01-25/17-37-25_aloha_act",
|
159 |
"job_name": "aloha_act",
|
160 |
"resume": false,
|
|
|
|
|
161 |
"seed": 1000,
|
162 |
"num_workers": 4,
|
163 |
"batch_size": 8,
|