Robotics
LeRobot
Safetensors
act
hacks101 commited on
Commit
a0ddbfa
·
verified ·
1 Parent(s): 47270e1

Upload policy weights, train config and readme

Browse files
Files changed (3) hide show
  1. README.md +2 -2
  2. model.safetensors +1 -1
  3. train_config.json +6 -6
README.md CHANGED
@@ -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
- lerobot-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
- lerobot-record \
50
  --robot.type=so100_follower \
51
  --dataset.repo_id=<hf_user>/eval_<dataset> \
52
  --policy.path=<hf_user>/<desired_policy_repo_id> \
 
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
  ### 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> \
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b39d0388fe7116356b42fef5209f61fab0611c8f12a285bae4293c1c349da982
3
  size 206700800
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8d03ee1909a338f4b58fe8ceb9dd8878e6c1f98bba1af4f55fb90378133dce14
3
  size 206700800
train_config.json CHANGED
@@ -126,13 +126,13 @@
126
  "optimizer_weight_decay": 0.0001,
127
  "optimizer_lr_backbone": 1e-05
128
  },
129
- "output_dir": "outputs/train/pick-a-bolts-300",
130
- "job_name": "act-pick-a-bolts",
131
  "resume": false,
132
  "seed": 1000,
133
- "num_workers": 4,
134
- "batch_size": 8,
135
- "steps": 100000,
136
  "eval_freq": 20000,
137
  "log_freq": 200,
138
  "save_checkpoint": true,
@@ -161,7 +161,7 @@
161
  "project": "lerobot",
162
  "entity": null,
163
  "notes": null,
164
- "run_id": "s8gqvi11",
165
  "mode": null
166
  }
167
  }
 
126
  "optimizer_weight_decay": 0.0001,
127
  "optimizer_lr_backbone": 1e-05
128
  },
129
+ "output_dir": "outputs/train/pick-a-bolts-300-700000",
130
+ "job_name": "act-pick-a-bolts-300-700000",
131
  "resume": false,
132
  "seed": 1000,
133
+ "num_workers": 8,
134
+ "batch_size": 64,
135
+ "steps": 700000,
136
  "eval_freq": 20000,
137
  "log_freq": 200,
138
  "save_checkpoint": true,
 
161
  "project": "lerobot",
162
  "entity": null,
163
  "notes": null,
164
+ "run_id": "ygyehyy4",
165
  "mode": null
166
  }
167
  }