Hugging Face's logo

lerobot
/
act_aloha_sim_insertion_human

Robotics
Safetensors
act-policy
model_hub_mixin
pytorch_model_hub_mixin
Model card Files Community
2

refactor/pretrainedconfig_device_amps_args

#2
by imstevenpmwork HF staff - opened
base: refs/heads/main
←
from: refs/pr/2
Discussion Files changed
+4
-2
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,