Neptune-T commited on
Commit
4ab9bec
Β·
1 Parent(s): 520b4be

Simplify checkpoint variant names

Browse files
This view is limited to 50 files because it contains too many changes. Β  See raw diff
Files changed (50) hide show
  1. README.md +13 -17
  2. checkpoints/{main_table/07_pica_ours β†’ dragmesh2}/object_12583/handle_1/policy.pth +0 -0
  3. checkpoints/{main_table/07_pica_ours β†’ dragmesh2}/object_45261/handle_7/policy.pth +0 -0
  4. checkpoints/{main_table/07_pica_ours β†’ dragmesh2}/object_45661/handle_3/policy.pth +0 -0
  5. checkpoints/{main_table/07_pica_ours β†’ dragmesh2}/object_45936/handle_1/policy.pth +0 -0
  6. checkpoints/{main_table/07_pica_ours β†’ dragmesh2}/object_46440/handle_5/policy.pth +0 -0
  7. checkpoints/{main_table/07_pica_ours β†’ dragmesh2}/object_48513/handle_2/policy.pth +0 -0
  8. checkpoints/{main_table/07_pica_ours β†’ dragmesh2}/object_7310/handle_1/policy.pth +0 -0
  9. checkpoints/{main_table/05_gla_no_aux β†’ gla}/object_12583/handle_1/policy.pth +0 -0
  10. checkpoints/{main_table/05_gla_no_aux β†’ gla}/object_45261/handle_7/policy.pth +0 -0
  11. checkpoints/{main_table/05_gla_no_aux β†’ gla}/object_45661/handle_3/policy.pth +0 -0
  12. checkpoints/{main_table/05_gla_no_aux β†’ gla}/object_45936/handle_1/policy.pth +0 -0
  13. checkpoints/{main_table/05_gla_no_aux β†’ gla}/object_46440/handle_5/policy.pth +0 -0
  14. checkpoints/{main_table/05_gla_no_aux β†’ gla}/object_48513/handle_2/policy.pth +0 -0
  15. checkpoints/{main_table/05_gla_no_aux β†’ gla}/object_7310/handle_1/policy.pth +0 -0
  16. checkpoints/{main_table/08_gru_ppo β†’ gru}/object_12583/handle_1/policy.pth +0 -0
  17. checkpoints/{main_table/08_gru_ppo β†’ gru}/object_45261/handle_7/policy.pth +0 -0
  18. checkpoints/{main_table/08_gru_ppo β†’ gru}/object_45661/handle_3/policy.pth +0 -0
  19. checkpoints/{main_table/08_gru_ppo β†’ gru}/object_45936/handle_1/policy.pth +0 -0
  20. checkpoints/{main_table/08_gru_ppo β†’ gru}/object_46440/handle_5/policy.pth +0 -0
  21. checkpoints/{main_table/08_gru_ppo β†’ gru}/object_48513/handle_2/policy.pth +0 -0
  22. checkpoints/{main_table/08_gru_ppo β†’ gru}/object_7310/handle_1/policy.pth +0 -0
  23. checkpoints/{main_table/04_flat_history_ppo β†’ history}/object_12583/handle_1/policy.pth +0 -0
  24. checkpoints/{main_table/04_flat_history_ppo β†’ history}/object_45261/handle_7/policy.pth +0 -0
  25. checkpoints/{main_table/04_flat_history_ppo β†’ history}/object_45661/handle_3/policy.pth +0 -0
  26. checkpoints/{main_table/04_flat_history_ppo β†’ history}/object_45936/handle_1/policy.pth +0 -0
  27. checkpoints/{main_table/04_flat_history_ppo β†’ history}/object_46440/handle_5/policy.pth +0 -0
  28. checkpoints/{main_table/04_flat_history_ppo β†’ history}/object_48513/handle_2/policy.pth +0 -0
  29. checkpoints/{main_table/04_flat_history_ppo β†’ history}/object_7310/handle_1/policy.pth +0 -0
  30. checkpoints/{main_table/06_pica_no_gla_aux_v2c β†’ pica}/object_12583/handle_1/policy.pth +0 -0
  31. checkpoints/{main_table/06_pica_no_gla_aux_v2c β†’ pica}/object_45261/handle_7/policy.pth +0 -0
  32. checkpoints/{main_table/06_pica_no_gla_aux_v2c β†’ pica}/object_45661/handle_3/policy.pth +0 -0
  33. checkpoints/{main_table/06_pica_no_gla_aux_v2c β†’ pica}/object_45936/handle_1/policy.pth +0 -0
  34. checkpoints/{main_table/06_pica_no_gla_aux_v2c β†’ pica}/object_46440/handle_5/policy.pth +0 -0
  35. checkpoints/{main_table/06_pica_no_gla_aux_v2c β†’ pica}/object_48513/handle_2/policy.pth +0 -0
  36. checkpoints/{main_table/06_pica_no_gla_aux_v2c β†’ pica}/object_7310/handle_1/policy.pth +0 -0
  37. checkpoints/{main_table/03_state_only_ppo β†’ state}/object_12583/handle_1/policy.pth +0 -0
  38. checkpoints/{main_table/03_state_only_ppo β†’ state}/object_45261/handle_7/policy.pth +0 -0
  39. checkpoints/{main_table/03_state_only_ppo β†’ state}/object_45661/handle_3/policy.pth +0 -0
  40. checkpoints/{main_table/03_state_only_ppo β†’ state}/object_45936/handle_1/policy.pth +0 -0
  41. checkpoints/{main_table/03_state_only_ppo β†’ state}/object_46440/handle_5/policy.pth +0 -0
  42. checkpoints/{main_table/03_state_only_ppo β†’ state}/object_48513/handle_2/policy.pth +0 -0
  43. checkpoints/{main_table/03_state_only_ppo β†’ state}/object_7310/handle_1/policy.pth +0 -0
  44. checkpoints/{main_table/09_transformer_ppo β†’ transformer}/object_12583/handle_1/policy.pth +0 -0
  45. checkpoints/{main_table/09_transformer_ppo β†’ transformer}/object_45261/handle_7/policy.pth +0 -0
  46. checkpoints/{main_table/09_transformer_ppo β†’ transformer}/object_45661/handle_3/policy.pth +0 -0
  47. checkpoints/{main_table/09_transformer_ppo β†’ transformer}/object_45936/handle_1/policy.pth +0 -0
  48. checkpoints/{main_table/09_transformer_ppo β†’ transformer}/object_46440/handle_5/policy.pth +0 -0
  49. checkpoints/{main_table/09_transformer_ppo β†’ transformer}/object_48513/handle_2/policy.pth +0 -0
  50. checkpoints/{main_table/09_transformer_ppo β†’ transformer}/object_7310/handle_1/policy.pth +0 -0
README.md CHANGED
@@ -25,11 +25,10 @@ included in this model repository.
25
 
26
  ```text
27
  checkpoints/
28
- main_table/
29
- <experiment>/
30
- object_<object_id>/
31
- handle_<part_id>/
32
- policy.pth
33
  model_manifest.jsonl
34
  ```
35
 
@@ -39,18 +38,15 @@ path, original relative path, object category, file size, and SHA-256 digest.
39
 
40
  ## Policy variants
41
 
42
- | Experiment | Variant |
43
  | --- | --- |
44
- | `03_state_only_ppo` | State-only PPO baseline |
45
- | `04_flat_history_ppo` | Flat-history PPO baseline |
46
- | `05_gla_no_aux` | GLA without the auxiliary objective |
47
- | `06_pica_no_gla_aux_v2c` | PICA without the GLA auxiliary objective (`v2c`) |
48
- | `07_pica_ours` | DragMesh-2 PICA policy |
49
- | `08_gru_ppo` | GRU PPO baseline |
50
- | `09_transformer_ppo` | Transformer PPO baseline |
51
-
52
- The numeric prefixes preserve the main-table experiment numbering used during
53
- evaluation.
54
 
55
  ## Evaluation tasks
56
 
@@ -86,7 +82,7 @@ import numpy as np
86
  import torch
87
 
88
  checkpoint_path = Path(
89
- "checkpoints/main_table/07_pica_ours/"
90
  "object_45661/handle_3/policy.pth"
91
  )
92
 
 
25
 
26
  ```text
27
  checkpoints/
28
+ <variant>/
29
+ object_<object_id>/
30
+ handle_<part_id>/
31
+ policy.pth
 
32
  model_manifest.jsonl
33
  ```
34
 
 
38
 
39
  ## Policy variants
40
 
41
+ | Name | Variant |
42
  | --- | --- |
43
+ | `state` | State-only PPO baseline |
44
+ | `history` | Flat-history PPO baseline |
45
+ | `gla` | GLA without the auxiliary objective |
46
+ | `pica` | PICA without the GLA auxiliary objective (`v2c`) |
47
+ | `dragmesh2` | DragMesh-2 PICA policy |
48
+ | `gru` | GRU PPO baseline |
49
+ | `transformer` | Transformer PPO baseline |
 
 
 
50
 
51
  ## Evaluation tasks
52
 
 
82
  import torch
83
 
84
  checkpoint_path = Path(
85
+ "checkpoints/dragmesh2/"
86
  "object_45661/handle_3/policy.pth"
87
  )
88
 
checkpoints/{main_table/07_pica_ours β†’ dragmesh2}/object_12583/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/07_pica_ours β†’ dragmesh2}/object_45261/handle_7/policy.pth RENAMED
File without changes
checkpoints/{main_table/07_pica_ours β†’ dragmesh2}/object_45661/handle_3/policy.pth RENAMED
File without changes
checkpoints/{main_table/07_pica_ours β†’ dragmesh2}/object_45936/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/07_pica_ours β†’ dragmesh2}/object_46440/handle_5/policy.pth RENAMED
File without changes
checkpoints/{main_table/07_pica_ours β†’ dragmesh2}/object_48513/handle_2/policy.pth RENAMED
File without changes
checkpoints/{main_table/07_pica_ours β†’ dragmesh2}/object_7310/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/05_gla_no_aux β†’ gla}/object_12583/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/05_gla_no_aux β†’ gla}/object_45261/handle_7/policy.pth RENAMED
File without changes
checkpoints/{main_table/05_gla_no_aux β†’ gla}/object_45661/handle_3/policy.pth RENAMED
File without changes
checkpoints/{main_table/05_gla_no_aux β†’ gla}/object_45936/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/05_gla_no_aux β†’ gla}/object_46440/handle_5/policy.pth RENAMED
File without changes
checkpoints/{main_table/05_gla_no_aux β†’ gla}/object_48513/handle_2/policy.pth RENAMED
File without changes
checkpoints/{main_table/05_gla_no_aux β†’ gla}/object_7310/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/08_gru_ppo β†’ gru}/object_12583/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/08_gru_ppo β†’ gru}/object_45261/handle_7/policy.pth RENAMED
File without changes
checkpoints/{main_table/08_gru_ppo β†’ gru}/object_45661/handle_3/policy.pth RENAMED
File without changes
checkpoints/{main_table/08_gru_ppo β†’ gru}/object_45936/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/08_gru_ppo β†’ gru}/object_46440/handle_5/policy.pth RENAMED
File without changes
checkpoints/{main_table/08_gru_ppo β†’ gru}/object_48513/handle_2/policy.pth RENAMED
File without changes
checkpoints/{main_table/08_gru_ppo β†’ gru}/object_7310/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/04_flat_history_ppo β†’ history}/object_12583/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/04_flat_history_ppo β†’ history}/object_45261/handle_7/policy.pth RENAMED
File without changes
checkpoints/{main_table/04_flat_history_ppo β†’ history}/object_45661/handle_3/policy.pth RENAMED
File without changes
checkpoints/{main_table/04_flat_history_ppo β†’ history}/object_45936/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/04_flat_history_ppo β†’ history}/object_46440/handle_5/policy.pth RENAMED
File without changes
checkpoints/{main_table/04_flat_history_ppo β†’ history}/object_48513/handle_2/policy.pth RENAMED
File without changes
checkpoints/{main_table/04_flat_history_ppo β†’ history}/object_7310/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/06_pica_no_gla_aux_v2c β†’ pica}/object_12583/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/06_pica_no_gla_aux_v2c β†’ pica}/object_45261/handle_7/policy.pth RENAMED
File without changes
checkpoints/{main_table/06_pica_no_gla_aux_v2c β†’ pica}/object_45661/handle_3/policy.pth RENAMED
File without changes
checkpoints/{main_table/06_pica_no_gla_aux_v2c β†’ pica}/object_45936/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/06_pica_no_gla_aux_v2c β†’ pica}/object_46440/handle_5/policy.pth RENAMED
File without changes
checkpoints/{main_table/06_pica_no_gla_aux_v2c β†’ pica}/object_48513/handle_2/policy.pth RENAMED
File without changes
checkpoints/{main_table/06_pica_no_gla_aux_v2c β†’ pica}/object_7310/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/03_state_only_ppo β†’ state}/object_12583/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/03_state_only_ppo β†’ state}/object_45261/handle_7/policy.pth RENAMED
File without changes
checkpoints/{main_table/03_state_only_ppo β†’ state}/object_45661/handle_3/policy.pth RENAMED
File without changes
checkpoints/{main_table/03_state_only_ppo β†’ state}/object_45936/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/03_state_only_ppo β†’ state}/object_46440/handle_5/policy.pth RENAMED
File without changes
checkpoints/{main_table/03_state_only_ppo β†’ state}/object_48513/handle_2/policy.pth RENAMED
File without changes
checkpoints/{main_table/03_state_only_ppo β†’ state}/object_7310/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/09_transformer_ppo β†’ transformer}/object_12583/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/09_transformer_ppo β†’ transformer}/object_45261/handle_7/policy.pth RENAMED
File without changes
checkpoints/{main_table/09_transformer_ppo β†’ transformer}/object_45661/handle_3/policy.pth RENAMED
File without changes
checkpoints/{main_table/09_transformer_ppo β†’ transformer}/object_45936/handle_1/policy.pth RENAMED
File without changes
checkpoints/{main_table/09_transformer_ppo β†’ transformer}/object_46440/handle_5/policy.pth RENAMED
File without changes
checkpoints/{main_table/09_transformer_ppo β†’ transformer}/object_48513/handle_2/policy.pth RENAMED
File without changes
checkpoints/{main_table/09_transformer_ppo β†’ transformer}/object_7310/handle_1/policy.pth RENAMED
File without changes