anhdao69 commited on
Commit
dbb527e
·
verified ·
1 Parent(s): e4715a1

Index: RoboTwin-10 seven-model release

Browse files
Files changed (3) hide show
  1. README.md +24 -1
  2. checkpoint_index.json +267 -1
  3. robotwin-10/README.md +20 -0
README.md CHANGED
@@ -9,7 +9,7 @@ tags:
9
 
10
  # LoopWAM final checkpoints
11
 
12
- Final trained policy weights from the LoopWAM experiments. This repository contains **25 checkpoints**, separated by training data scope. Both original and repeated 4/4 runs are retained. All production checkpoints are from completed ten-epoch runs with global batch 128 and training seed 42.
13
 
14
  ## October 9 additions: all-loop KV
15
 
@@ -93,10 +93,33 @@ v0 (video 4 / action 4) retrained on all four suites with the same data, normali
93
 
94
  Aligned (late) 4/2 v0 trained on all four suites with the same data, normalization, initialization recipe, schedule, precision and 2-GPU microbatch-8 layout as `libero-all-suites/v0-video4-action1` (21,700 updates, global batch 128, seed 42) on two H100s. Uses the LoopWAM_NT branch (revision `56e3328`). Not evaluated.
95
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
96
  ## Checkpoint index
97
 
98
  | Directory | Model | Video/action loops | Parameters | Final evaluation |
99
  |---|---|---|---:|---|
 
 
 
 
 
 
 
100
  | [libero-all-suites/init_variants/core_mean](libero-all-suites/init_variants/core_mean) | v0 init `core_mean` | 4/4 | 584,536,135 | Training only; evaluation not run |
101
  | [libero-all-suites/v0-video4-action2](libero-all-suites/v0-video4-action2) | v0 aligned | 4/2 | 584,536,135 | Training only; evaluation not run |
102
  | [libero-all-suites/init_variants/core16_21](libero-all-suites/init_variants/core16_21) | v0 init `core16_21` | 4/4 | 584,536,135 | Training only; evaluation not run |
 
9
 
10
  # LoopWAM final checkpoints
11
 
12
+ Final trained policy weights from the LoopWAM experiments. This repository contains **32 checkpoints**, separated by training data scope. Both original and repeated 4/4 runs are retained. All production checkpoints are from completed ten-epoch runs with global batch 128 and training seed 42.
13
 
14
  ## October 9 additions: all-loop KV
15
 
 
93
 
94
  Aligned (late) 4/2 v0 trained on all four suites with the same data, normalization, initialization recipe, schedule, precision and 2-GPU microbatch-8 layout as `libero-all-suites/v0-video4-action1` (21,700 updates, global batch 128, seed 42) on two H100s. Uses the LoopWAM_NT branch (revision `56e3328`). Not evaluated.
95
 
96
+ ## November 2026 addition: RoboTwin-10 (LiLa-WAM 10-task subset)
97
+
98
+ Seven LoopWAM architectures trained for five epochs on the 10-task RoboTwin-10 subset with the same data order, noise, initialization recipe and schedule; see [robotwin-10/README.md](robotwin-10/README.md).
99
+
100
+ | Directory | Model | Video/action loops | Parameters | Training time | Final loss (last 100) | Evaluation |
101
+ |---|---|---|---:|---:|---:|---|
102
+ | [robotwin-10/v0-video4-action4](robotwin-10/v0-video4-action4) | v0 | 4/4 | 584,567,886 | 17.8 h | 0.0453 | Training only; evaluation not run |
103
+ | [robotwin-10/dense-s12](robotwin-10/dense-s12) | dense_s12 | 1/1 | 584,567,886 | 8.0 h | 0.0485 | Training only; evaluation not run |
104
+ | [robotwin-10/dense-s30](robotwin-10/dense-s30) | dense_s30 | 1/1 | 1,416,145,998 | 19.4 h | 0.0350 | Training only; evaluation not run |
105
+ | [robotwin-10/v0-video4-action2](robotwin-10/v0-video4-action2) | v0 aligned | 4/2 | 584,567,886 | 16.7 h | 0.0452 | Training only; evaluation not run |
106
+ | [robotwin-10/v0-video4-action1](robotwin-10/v0-video4-action1) | v0 aligned | 4/1 | 584,567,886 | 16.3 h | 0.0453 | Training only; evaluation not run |
107
+ | [robotwin-10/v0-video2-action2](robotwin-10/v0-video2-action2) | v0 aligned | 2/2 | 584,567,886 | 11.1 h | 0.0469 | Training only; evaluation not run |
108
+ | [robotwin-10/v0-video1-action4](robotwin-10/v0-video1-action4) | v0 aligned | 1/4 | 584,567,886 | 11.9 h | 0.0485 | Training only; evaluation not run |
109
+
110
+ Training only; evaluation not run.
111
+
112
  ## Checkpoint index
113
 
114
  | Directory | Model | Video/action loops | Parameters | Final evaluation |
115
  |---|---|---|---:|---|
116
+ | [robotwin-10/v0-video4-action4](robotwin-10/v0-video4-action4) | v0 (RoboTwin-10) | 4/4 | 584,567,886 | Training only; evaluation not run |
117
+ | [robotwin-10/dense-s12](robotwin-10/dense-s12) | dense_s12 (RoboTwin-10) | 1/1 | 584,567,886 | Training only; evaluation not run |
118
+ | [robotwin-10/dense-s30](robotwin-10/dense-s30) | dense_s30 (RoboTwin-10) | 1/1 | 1,416,145,998 | Training only; evaluation not run |
119
+ | [robotwin-10/v0-video4-action2](robotwin-10/v0-video4-action2) | v0 aligned (RoboTwin-10) | 4/2 | 584,567,886 | Training only; evaluation not run |
120
+ | [robotwin-10/v0-video4-action1](robotwin-10/v0-video4-action1) | v0 aligned (RoboTwin-10) | 4/1 | 584,567,886 | Training only; evaluation not run |
121
+ | [robotwin-10/v0-video2-action2](robotwin-10/v0-video2-action2) | v0 aligned (RoboTwin-10) | 2/2 | 584,567,886 | Training only; evaluation not run |
122
+ | [robotwin-10/v0-video1-action4](robotwin-10/v0-video1-action4) | v0 aligned (RoboTwin-10) | 1/4 | 584,567,886 | Training only; evaluation not run |
123
  | [libero-all-suites/init_variants/core_mean](libero-all-suites/init_variants/core_mean) | v0 init `core_mean` | 4/4 | 584,536,135 | Training only; evaluation not run |
124
  | [libero-all-suites/v0-video4-action2](libero-all-suites/v0-video4-action2) | v0 aligned | 4/2 | 584,536,135 | Training only; evaluation not run |
125
  | [libero-all-suites/init_variants/core16_21](libero-all-suites/init_variants/core16_21) | v0 init `core16_21` | 4/4 | 584,536,135 | Training only; evaluation not run |
checkpoint_index.json CHANGED
@@ -788,7 +788,273 @@
788
  "microbatch": 8,
789
  "gradient_accumulation": 8,
790
  "created_utc": "2026-10-28T19:14:54.782633+00:00"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
791
  }
792
  ],
793
- "updated_utc": "2026-10-28T19:18:44.471588+00:00"
794
  }
 
788
  "microbatch": 8,
789
  "gradient_accumulation": 8,
790
  "created_utc": "2026-10-28T19:14:54.782633+00:00"
791
+ },
792
+ {
793
+ "id": "robotwin10_v0_video4_action4",
794
+ "repo_path": "robotwin-10/v0-video4-action4",
795
+ "training_path": "/groups/yshang/an221229/checkpoints/LoopWAM_NT/robotwin10/v0-video4-action4/train",
796
+ "export_sha256": "bb1563f16bcd102b3f4ad4a1a14377ee49de7e461d862b7bbc5dba497f036127",
797
+ "export_bytes": 2338751845,
798
+ "step": 45445,
799
+ "format": "loopwam-s-robot-v1",
800
+ "optimizer_included": false,
801
+ "version": "v0",
802
+ "video_loops": 4,
803
+ "action_loops": 4,
804
+ "loop_alignment": "late",
805
+ "parameters": 584567886,
806
+ "dataset_scope": "robotwin10",
807
+ "tasks": [
808
+ "adjust_bottle",
809
+ "grab_roller",
810
+ "hanging_mug",
811
+ "move_stapler_pad",
812
+ "open_microwave",
813
+ "press_stapler",
814
+ "scan_object",
815
+ "stack_blocks_two",
816
+ "stamp_seal",
817
+ "turn_switch"
818
+ ],
819
+ "epochs": 5,
820
+ "global_batch": 128,
821
+ "training_seed": 42,
822
+ "initialization_mode": "fresh_wan_donors_no_embodied_pretraining",
823
+ "evaluation_status": "not_evaluated",
824
+ "training_seconds": 64081.80207467079,
825
+ "final_loss_last100": 0.045258548930287364,
826
+ "tensor_equality_verified": true,
827
+ "source_revision": "bea472670ea52bb375ae75ba5380cd88087d2c30",
828
+ "hf_revision": "df3d575d3f3fde3cf1b8d5f609fab93dc6a95649"
829
+ },
830
+ {
831
+ "id": "robotwin10_dense_s12",
832
+ "repo_path": "robotwin-10/dense-s12",
833
+ "training_path": "/groups/yshang/an221229/checkpoints/LoopWAM_NT/robotwin10/dense-s12/train",
834
+ "export_sha256": "c0c0fbb7b4cc607c00f7ee783c1f0e43815daa9fa09b1141c0b456dec255d064",
835
+ "export_bytes": 2338751909,
836
+ "step": 45445,
837
+ "format": "loopwam-s-robot-v1",
838
+ "optimizer_included": false,
839
+ "version": "dense_s12",
840
+ "video_loops": 1,
841
+ "action_loops": 1,
842
+ "loop_alignment": "late",
843
+ "parameters": 584567886,
844
+ "dataset_scope": "robotwin10",
845
+ "tasks": [
846
+ "adjust_bottle",
847
+ "grab_roller",
848
+ "hanging_mug",
849
+ "move_stapler_pad",
850
+ "open_microwave",
851
+ "press_stapler",
852
+ "scan_object",
853
+ "stack_blocks_two",
854
+ "stamp_seal",
855
+ "turn_switch"
856
+ ],
857
+ "epochs": 5,
858
+ "global_batch": 128,
859
+ "training_seed": 42,
860
+ "initialization_mode": "fresh_wan_donors_no_embodied_pretraining",
861
+ "evaluation_status": "not_evaluated",
862
+ "training_seconds": 28743.092683792114,
863
+ "final_loss_last100": 0.0485357765853405,
864
+ "tensor_equality_verified": true,
865
+ "source_revision": "bea472670ea52bb375ae75ba5380cd88087d2c30",
866
+ "hf_revision": "828d6802c684d44b834a656367faeb1858abea74"
867
+ },
868
+ {
869
+ "id": "robotwin10_dense_s30",
870
+ "repo_path": "robotwin-10/dense-s30",
871
+ "training_path": "/groups/yshang/an221229/checkpoints/LoopWAM_NT/robotwin10/dense-s30/train",
872
+ "export_sha256": "cc549b65bb849198ab1d11fd78e2a14bd0d1682927665b24c5f82b3ee8b78985",
873
+ "export_bytes": 5665462576,
874
+ "step": 45445,
875
+ "format": "loopwam-s-robot-v1",
876
+ "optimizer_included": false,
877
+ "version": "dense_s30",
878
+ "video_loops": 1,
879
+ "action_loops": 1,
880
+ "loop_alignment": "late",
881
+ "parameters": 1416145998,
882
+ "dataset_scope": "robotwin10",
883
+ "tasks": [
884
+ "adjust_bottle",
885
+ "grab_roller",
886
+ "hanging_mug",
887
+ "move_stapler_pad",
888
+ "open_microwave",
889
+ "press_stapler",
890
+ "scan_object",
891
+ "stack_blocks_two",
892
+ "stamp_seal",
893
+ "turn_switch"
894
+ ],
895
+ "epochs": 5,
896
+ "global_batch": 128,
897
+ "training_seed": 42,
898
+ "initialization_mode": "fresh_wan_donors_no_embodied_pretraining",
899
+ "evaluation_status": "not_evaluated",
900
+ "training_seconds": 69898.21527791023,
901
+ "final_loss_last100": 0.03495093105360866,
902
+ "tensor_equality_verified": true,
903
+ "source_revision": "bea472670ea52bb375ae75ba5380cd88087d2c30",
904
+ "hf_revision": "cdf71976208ade41a93465f6b8166b5e6b7170ca"
905
+ },
906
+ {
907
+ "id": "robotwin10_v0_video4_action2",
908
+ "repo_path": "robotwin-10/v0-video4-action2",
909
+ "training_path": "/groups/yshang/an221229/checkpoints/LoopWAM_NT/robotwin10/v0-video4-action2/train",
910
+ "export_sha256": "fea2ce45c09197fe68df9345d1f60e520dab8193dd7ee9b5147001e64efc32c9",
911
+ "export_bytes": 2338751845,
912
+ "step": 45445,
913
+ "format": "loopwam-s-robot-v1",
914
+ "optimizer_included": false,
915
+ "version": "v0",
916
+ "video_loops": 4,
917
+ "action_loops": 2,
918
+ "loop_alignment": "late",
919
+ "parameters": 584567886,
920
+ "dataset_scope": "robotwin10",
921
+ "tasks": [
922
+ "adjust_bottle",
923
+ "grab_roller",
924
+ "hanging_mug",
925
+ "move_stapler_pad",
926
+ "open_microwave",
927
+ "press_stapler",
928
+ "scan_object",
929
+ "stack_blocks_two",
930
+ "stamp_seal",
931
+ "turn_switch"
932
+ ],
933
+ "epochs": 5,
934
+ "global_batch": 128,
935
+ "training_seed": 42,
936
+ "initialization_mode": "fresh_wan_donors_no_embodied_pretraining",
937
+ "evaluation_status": "not_evaluated",
938
+ "training_seconds": 60219.636355638504,
939
+ "final_loss_last100": 0.045222468487918374,
940
+ "tensor_equality_verified": true,
941
+ "source_revision": "bea472670ea52bb375ae75ba5380cd88087d2c30",
942
+ "hf_revision": "b24d5ed114c5d812284472c674bee31470a8d8ff"
943
+ },
944
+ {
945
+ "id": "robotwin10_v0_video4_action1",
946
+ "repo_path": "robotwin-10/v0-video4-action1",
947
+ "training_path": "/groups/yshang/an221229/checkpoints/LoopWAM_NT/robotwin10/v0-video4-action1/train",
948
+ "export_sha256": "b0fd9da51d9ff5ab512d5729de6565e85d240a10f078ddf9f803733a77aa757d",
949
+ "export_bytes": 2338751909,
950
+ "step": 45445,
951
+ "format": "loopwam-s-robot-v1",
952
+ "optimizer_included": false,
953
+ "version": "v0",
954
+ "video_loops": 4,
955
+ "action_loops": 1,
956
+ "loop_alignment": "late",
957
+ "parameters": 584567886,
958
+ "dataset_scope": "robotwin10",
959
+ "tasks": [
960
+ "adjust_bottle",
961
+ "grab_roller",
962
+ "hanging_mug",
963
+ "move_stapler_pad",
964
+ "open_microwave",
965
+ "press_stapler",
966
+ "scan_object",
967
+ "stack_blocks_two",
968
+ "stamp_seal",
969
+ "turn_switch"
970
+ ],
971
+ "epochs": 5,
972
+ "global_batch": 128,
973
+ "training_seed": 42,
974
+ "initialization_mode": "fresh_wan_donors_no_embodied_pretraining",
975
+ "evaluation_status": "not_evaluated",
976
+ "training_seconds": 58643.347306489944,
977
+ "final_loss_last100": 0.045315117612481116,
978
+ "tensor_equality_verified": true,
979
+ "source_revision": "bea472670ea52bb375ae75ba5380cd88087d2c30",
980
+ "hf_revision": "ba2d4a5e921e2e2e298ea55d811f421932f607b8"
981
+ },
982
+ {
983
+ "id": "robotwin10_v0_video2_action2",
984
+ "repo_path": "robotwin-10/v0-video2-action2",
985
+ "training_path": "/groups/yshang/an221229/checkpoints/LoopWAM_NT/robotwin10/v0-video2-action2/train",
986
+ "export_sha256": "aa835dce6898f92d49275bf6e6b4e8f9202d4a8025a1e810ae3158867369be59",
987
+ "export_bytes": 2338751845,
988
+ "step": 45445,
989
+ "format": "loopwam-s-robot-v1",
990
+ "optimizer_included": false,
991
+ "version": "v0",
992
+ "video_loops": 2,
993
+ "action_loops": 2,
994
+ "loop_alignment": "late",
995
+ "parameters": 584567886,
996
+ "dataset_scope": "robotwin10",
997
+ "tasks": [
998
+ "adjust_bottle",
999
+ "grab_roller",
1000
+ "hanging_mug",
1001
+ "move_stapler_pad",
1002
+ "open_microwave",
1003
+ "press_stapler",
1004
+ "scan_object",
1005
+ "stack_blocks_two",
1006
+ "stamp_seal",
1007
+ "turn_switch"
1008
+ ],
1009
+ "epochs": 5,
1010
+ "global_batch": 128,
1011
+ "training_seed": 42,
1012
+ "initialization_mode": "fresh_wan_donors_no_embodied_pretraining",
1013
+ "evaluation_status": "not_evaluated",
1014
+ "training_seconds": 40010.85231876373,
1015
+ "final_loss_last100": 0.04690537687391043,
1016
+ "tensor_equality_verified": true,
1017
+ "source_revision": "bea472670ea52bb375ae75ba5380cd88087d2c30",
1018
+ "hf_revision": "34af15d8b3fa65c1938c48e806599df29da848d2"
1019
+ },
1020
+ {
1021
+ "id": "robotwin10_v0_video1_action4",
1022
+ "repo_path": "robotwin-10/v0-video1-action4",
1023
+ "training_path": "/groups/yshang/an221229/checkpoints/LoopWAM_NT/robotwin10/v0-video1-action4/train",
1024
+ "export_sha256": "f9b097164b1fb076a0440f7c42e6439ee87885fbcf21d88975ae3f2d44fd1daf",
1025
+ "export_bytes": 2338751909,
1026
+ "step": 45445,
1027
+ "format": "loopwam-s-robot-v1",
1028
+ "optimizer_included": false,
1029
+ "version": "v0",
1030
+ "video_loops": 1,
1031
+ "action_loops": 4,
1032
+ "loop_alignment": "late",
1033
+ "parameters": 584567886,
1034
+ "dataset_scope": "robotwin10",
1035
+ "tasks": [
1036
+ "adjust_bottle",
1037
+ "grab_roller",
1038
+ "hanging_mug",
1039
+ "move_stapler_pad",
1040
+ "open_microwave",
1041
+ "press_stapler",
1042
+ "scan_object",
1043
+ "stack_blocks_two",
1044
+ "stamp_seal",
1045
+ "turn_switch"
1046
+ ],
1047
+ "epochs": 5,
1048
+ "global_batch": 128,
1049
+ "training_seed": 42,
1050
+ "initialization_mode": "fresh_wan_donors_no_embodied_pretraining",
1051
+ "evaluation_status": "not_evaluated",
1052
+ "training_seconds": 42788.281959056854,
1053
+ "final_loss_last100": 0.0485189451277256,
1054
+ "tensor_equality_verified": true,
1055
+ "source_revision": "bea472670ea52bb375ae75ba5380cd88087d2c30",
1056
+ "hf_revision": "e4715a1148c88570090c5d470243c4e60701c2bb"
1057
  }
1058
  ],
1059
+ "updated_utc": "2026-11-05T11:35:03.567352+00:00"
1060
  }
robotwin-10/README.md ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # RoboTwin-10 checkpoints
2
+
3
+ Seven LoopWAM models trained on **RoboTwin-10**, the 10-task LiLa-WAM ablation subset of the FastWAM RoboTwin 2.0
4
+ release (adjust_bottle, grab_roller, hanging_mug, move_stapler_pad, open_microwave, press_stapler, scan_object, stack_blocks_two, stamp_seal, turn_switch; 50 clean + 500 randomized demonstrations per task, 5,500 episodes,
5
+ 1,163,351 windows per epoch). Every model uses the same recipe as `robotwin/v0-video4-action4` restricted to these
6
+ tasks: fresh canonical Wan2.1 donors plus seeded 14-D robot interfaces, exactly 128 windows per update, five epochs
7
+ (45,445 updates), seed 42, identical data order and per-window noise, AdamW 1e-4 with 5% warmup and cosine to
8
+ 1e-6. Only the architecture differs. Each folder has its own README, config, provenance, metrics and SHA256SUMS.
9
+
10
+ | Directory | Model | Video/action loops | Parameters | Training time | Final loss (last 100) | Evaluation |
11
+ |---|---|---|---:|---:|---:|---|
12
+ | [robotwin-10/v0-video4-action4](robotwin-10/v0-video4-action4) | v0 | 4/4 | 584,567,886 | 17.8 h | 0.0453 | Training only; evaluation not run |
13
+ | [robotwin-10/dense-s12](robotwin-10/dense-s12) | dense_s12 | 1/1 | 584,567,886 | 8.0 h | 0.0485 | Training only; evaluation not run |
14
+ | [robotwin-10/dense-s30](robotwin-10/dense-s30) | dense_s30 | 1/1 | 1,416,145,998 | 19.4 h | 0.0350 | Training only; evaluation not run |
15
+ | [robotwin-10/v0-video4-action2](robotwin-10/v0-video4-action2) | v0 aligned | 4/2 | 584,567,886 | 16.7 h | 0.0452 | Training only; evaluation not run |
16
+ | [robotwin-10/v0-video4-action1](robotwin-10/v0-video4-action1) | v0 aligned | 4/1 | 584,567,886 | 16.3 h | 0.0453 | Training only; evaluation not run |
17
+ | [robotwin-10/v0-video2-action2](robotwin-10/v0-video2-action2) | v0 aligned | 2/2 | 584,567,886 | 11.1 h | 0.0469 | Training only; evaluation not run |
18
+ | [robotwin-10/v0-video1-action4](robotwin-10/v0-video1-action4) | v0 aligned | 1/4 | 584,567,886 | 11.9 h | 0.0485 | Training only; evaluation not run |
19
+
20
+ **Not evaluated:** training-only releases; no RoboTwin success rate is claimed.