Robotics
LeRobot
Safetensors
act
fitrijamat commited on
Commit
f9562cd
·
verified ·
1 Parent(s): 804ecce

Upload policy weights, train config and readme

Browse files
Files changed (4) hide show
  1. README.md +3 -3
  2. config.json +1 -0
  3. model.safetensors +2 -2
  4. train_config.json +5 -3
README.md CHANGED
@@ -6,8 +6,8 @@ model_name: act
6
  pipeline_tag: robotics
7
  tags:
8
  - robotics
9
- - act
10
  - lerobot
 
11
  ---
12
 
13
  # Model Card for act
@@ -31,7 +31,7 @@ Below is the short version on how to train and run inference/eval:
31
  ### Train from scratch
32
 
33
  ```bash
34
- python -m lerobot.scripts.train \
35
  --dataset.repo_id=${HF_USER}/<dataset> \
36
  --policy.type=act \
37
  --output_dir=outputs/train/<desired_policy_repo_id> \
@@ -46,7 +46,7 @@ _Writes checkpoints to `outputs/train/<desired_policy_repo_id>/checkpoints/`._
46
  ### Evaluate the policy/run inference
47
 
48
  ```bash
49
- python -m lerobot.record \
50
  --robot.type=so100_follower \
51
  --dataset.repo_id=<hf_user>/eval_<dataset> \
52
  --policy.path=<hf_user>/<desired_policy_repo_id> \
 
6
  pipeline_tag: robotics
7
  tags:
8
  - robotics
 
9
  - lerobot
10
+ - act
11
  ---
12
 
13
  # Model Card for act
 
31
  ### Train from scratch
32
 
33
  ```bash
34
+ lerobot-train \
35
  --dataset.repo_id=${HF_USER}/<dataset> \
36
  --policy.type=act \
37
  --output_dir=outputs/train/<desired_policy_repo_id> \
 
46
  ### Evaluate the policy/run inference
47
 
48
  ```bash
49
+ lerobot-record \
50
  --robot.type=so100_follower \
51
  --dataset.repo_id=<hf_user>/eval_<dataset> \
52
  --policy.path=<hf_user>/<desired_policy_repo_id> \
config.json CHANGED
@@ -40,6 +40,7 @@
40
  },
41
  "device": "cuda",
42
  "use_amp": false,
 
43
  "push_to_hub": true,
44
  "repo_id": "fitrijamat/pick-insert-block",
45
  "private": null,
 
40
  },
41
  "device": "cuda",
42
  "use_amp": false,
43
+ "gradient_accumulation_steps": 1,
44
  "push_to_hub": true,
45
  "repo_id": "fitrijamat/pick-insert-block",
46
  "private": null,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c8cb46bf970338cb5c3899173316106214f96834a7a60fd03caba08d1a2abbad
3
- size 206701064
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:00c3c9df6ba6ddaa7fa2623da41cc0b27639c401f0be7c2dad2d994f4380288d
3
+ size 206699736
train_config.json CHANGED
@@ -62,7 +62,8 @@
62
  },
63
  "revision": "main",
64
  "use_imagenet_stats": true,
65
- "video_backend": "pyav"
 
66
  },
67
  "env": null,
68
  "policy": {
@@ -107,6 +108,7 @@
107
  },
108
  "device": "cuda",
109
  "use_amp": false,
 
110
  "push_to_hub": true,
111
  "repo_id": "fitrijamat/pick-insert-block",
112
  "private": null,
@@ -134,7 +136,7 @@
134
  "optimizer_weight_decay": 0.0001,
135
  "optimizer_lr_backbone": 1e-05
136
  },
137
- "output_dir": "C:\\Users\\User\\Documents\\reuben_ws\\lerobot\\outputs\\train\\act_mini_studio_block_in_cup_new",
138
  "job_name": "act_pick-insert-block",
139
  "resume": false,
140
  "seed": 1000,
@@ -169,7 +171,7 @@
169
  "project": "lerobot",
170
  "entity": null,
171
  "notes": null,
172
- "run_id": "k482szox",
173
  "mode": null
174
  }
175
  }
 
62
  },
63
  "revision": "main",
64
  "use_imagenet_stats": true,
65
+ "video_backend": "pyav",
66
+ "streaming": false
67
  },
68
  "env": null,
69
  "policy": {
 
108
  },
109
  "device": "cuda",
110
  "use_amp": false,
111
+ "gradient_accumulation_steps": 1,
112
  "push_to_hub": true,
113
  "repo_id": "fitrijamat/pick-insert-block",
114
  "private": null,
 
136
  "optimizer_weight_decay": 0.0001,
137
  "optimizer_lr_backbone": 1e-05
138
  },
139
+ "output_dir": "C:\\Users\\User\\Documents\\reuben_ws\\lerobot\\outputs\\train\\act_mini_studio_block_in_cup_latest",
140
  "job_name": "act_pick-insert-block",
141
  "resume": false,
142
  "seed": 1000,
 
171
  "project": "lerobot",
172
  "entity": null,
173
  "notes": null,
174
+ "run_id": "edc35ocv",
175
  "mode": null
176
  }
177
  }