Transformers
Safetensors
Inference Endpoints
alexandersoare commited on
Commit
d6c8f71
1 Parent(s): 5b65dc6

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +3 -1
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.