imstevenpmwork HF staff commited on
Commit
356e503
·
verified ·
1 Parent(s): 2bd896d

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. config.json +2 -0
  2. 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,