Transformers
Safetensors
Inference Endpoints
alexandersoare commited on
Commit
5b6ce85
1 Parent(s): 8502e35

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +3 -4
README.md CHANGED
@@ -19,9 +19,7 @@ Trained with [LeRobot@d747195](https://github.com/huggingface/lerobot/tree/d7471
19
 
20
  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.
21
 
22
- Here are the [loss](./train_loss.csv) and [evaluation success rate](./eval_pc_success.csv) (with 50 rollouts) during training.
23
-
24
- ![](training_curves.png)
25
 
26
  This took about 2.5 hours to train on an Nvida RTX 3090.
27
 
@@ -29,7 +27,8 @@ This took about 2.5 hours to train on an Nvida RTX 3090.
29
 
30
  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.
31
 
32
- Here are the success rate results for 500 episodes worth of evaluation. The first row is the naive mean. The second row assumes a uniform prior and computes the beta posterior, then calculates the mean and lower/upper confidence bounds (with a 68.2% confidence interval centered on the mean). The "Theirs" column is for an equivalent model trained on the original ACT repository and evaluated on LeRobot (the model weights may be found in the [`original_act_repo`](https://huggingface.co/lerobot/act_aloha_sim_transfer_cube_human/tree/original_act_repo) branch of this respository).
 
33
 
34
  <blank>|Ours|Theirs
35
  -|-|-
 
19
 
20
  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.
21
 
22
+ The training curves may be found at https://wandb.ai/alexander-soare/Alexander-LeRobot/runs/hjdard15.
 
 
23
 
24
  This took about 2.5 hours to train on an Nvida RTX 3090.
25
 
 
27
 
28
  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.
29
 
30
+ Here are the success rate results for 500 episodes worth of evaluation. The first row is the naive mean. The second row assumes a uniform prior and computes the beta posterior, then calculates the mean and lower/upper confidence bounds (with a 68.2% confidence interval centered on the mean). The "Theirs" column is for an equivalent model trained on the original ACT repository and evaluated on LeRobot (the model weights may be found in the [`original_act_repo`](https://huggingface.co/lerobot/act_aloha_sim_transfer_cube_human/tree/original_act_repo) branch of this respository). The results of each of the individual rollouts may be found in [eval_info.json](eval_info.json).
31
+
32
 
33
  <blank>|Ours|Theirs
34
  -|-|-