alexandersoare
commited on
Commit
•
d6c8f71
1
Parent(s):
5b65dc6
Upload folder using huggingface_hub
Browse files
README.md
CHANGED
@@ -10,7 +10,7 @@ See the [LeRobot library](https://github.com/huggingface/lerobot) (particularly
|
|
10 |
|
11 |
## Training Details
|
12 |
|
13 |
-
Trained with [LeRobot@d747195](https://github.com/huggingface/lerobot/tree/d747195c5733c4f68d4bfbe62632d6fc1b605712)
|
14 |
|
15 |
The model was trained using [LeRobot's training script](https://github.com/huggingface/lerobot/blob/d747195c5733c4f68d4bfbe62632d6fc1b605712/lerobot/scripts/train.py) and with the [aloha_sim_transfer_cube_human](https://huggingface.co/datasets/lerobot/aloha_sim_transfer_cube_human/tree/v1.3) dataset.
|
16 |
|
@@ -18,6 +18,8 @@ Here are the [loss](./train_loss.csv) and [evaluation success rate](./eval_pc_su
|
|
18 |
|
19 |
![](training_curves.png)
|
20 |
|
|
|
|
|
21 |
## Evaluation
|
22 |
|
23 |
The model was evaluated on the `AlohaTransferCube` environment from [gym-aloha](https://github.com/huggingface/gym-aloha) and compared to a similar model trained with the original [ACT repository](https://github.com/tonyzhaozh/act). Each episode marks a success if the cube is successfully picked by one robot arm and transferred to the other robot arm.
|
|
|
10 |
|
11 |
## Training Details
|
12 |
|
13 |
+
Trained with [LeRobot@d747195](https://github.com/huggingface/lerobot/tree/d747195c5733c4f68d4bfbe62632d6fc1b605712).
|
14 |
|
15 |
The model was trained using [LeRobot's training script](https://github.com/huggingface/lerobot/blob/d747195c5733c4f68d4bfbe62632d6fc1b605712/lerobot/scripts/train.py) and with the [aloha_sim_transfer_cube_human](https://huggingface.co/datasets/lerobot/aloha_sim_transfer_cube_human/tree/v1.3) dataset.
|
16 |
|
|
|
18 |
|
19 |
![](training_curves.png)
|
20 |
|
21 |
+
This took about 2.5 hours to train on an Nvida RTX 3090.
|
22 |
+
|
23 |
## Evaluation
|
24 |
|
25 |
The model was evaluated on the `AlohaTransferCube` environment from [gym-aloha](https://github.com/huggingface/gym-aloha) and compared to a similar model trained with the original [ACT repository](https://github.com/tonyzhaozh/act). Each episode marks a success if the cube is successfully picked by one robot arm and transferred to the other robot arm.
|