Robotics
LeRobot
Safetensors
act
Hardwarize commited on
Commit
d5ff5ca
·
verified ·
1 Parent(s): 9752446

Upload policy weights, train config and readme

Browse files
Files changed (4) hide show
  1. README.md +1 -1
  2. config.json +1 -1
  3. model.safetensors +1 -1
  4. train_config.json +2 -2
README.md CHANGED
@@ -5,9 +5,9 @@ license: apache-2.0
5
  model_name: act
6
  pipeline_tag: robotics
7
  tags:
 
8
  - robotics
9
  - lerobot
10
- - act
11
  ---
12
 
13
  # Model Card for act
 
5
  model_name: act
6
  pipeline_tag: robotics
7
  tags:
8
+ - act
9
  - robotics
10
  - lerobot
 
11
  ---
12
 
13
  # Model Card for act
config.json CHANGED
@@ -38,7 +38,7 @@
38
  ]
39
  }
40
  },
41
- "device": "cuda",
42
  "use_amp": false,
43
  "push_to_hub": true,
44
  "repo_id": "Hardwarize/first_policy1",
 
38
  ]
39
  }
40
  },
41
+ "device": "cpu",
42
  "use_amp": false,
43
  "push_to_hub": true,
44
  "repo_id": "Hardwarize/first_policy1",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c00bb228ebaa059137bdad396771e08c4fbae648492bb9601717ad021f296dde
3
  size 206742080
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d91561ab98a2bf0b1f31894a6b69079997621c709cdf250d561ac73fe8e66215
3
  size 206742080
train_config.json CHANGED
@@ -105,7 +105,7 @@
105
  ]
106
  }
107
  },
108
- "device": "cuda",
109
  "use_amp": false,
110
  "push_to_hub": true,
111
  "repo_id": "Hardwarize/first_policy1",
@@ -140,7 +140,7 @@
140
  "seed": 1000,
141
  "num_workers": 4,
142
  "batch_size": 8,
143
- "steps": 50000,
144
  "eval_freq": 20000,
145
  "log_freq": 200,
146
  "save_checkpoint": true,
 
105
  ]
106
  }
107
  },
108
+ "device": "cpu",
109
  "use_amp": false,
110
  "push_to_hub": true,
111
  "repo_id": "Hardwarize/first_policy1",
 
140
  "seed": 1000,
141
  "num_workers": 4,
142
  "batch_size": 8,
143
+ "steps": 1,
144
  "eval_freq": 20000,
145
  "log_freq": 200,
146
  "save_checkpoint": true,