yfan07 commited on
Commit
214e208
·
verified ·
1 Parent(s): 65fb4ac

Add files using upload-large-folder tool

Browse files
Files changed (35) hide show
  1. configs/config.py +14 -0
  2. data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00000.png +0 -0
  3. data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00001.png +0 -0
  4. data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00002.png +0 -0
  5. data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00003.png +0 -0
  6. data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00004.png +0 -0
  7. data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00005.png +0 -0
  8. data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00006.png +0 -0
  9. data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00007.png +0 -0
  10. data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00008.png +0 -0
  11. data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00009.png +0 -0
  12. data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00000.png +0 -0
  13. data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00001.png +0 -0
  14. data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00002.png +0 -0
  15. data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00003.png +0 -0
  16. data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00004.png +0 -0
  17. data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00005.png +0 -0
  18. data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00006.png +0 -0
  19. data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00007.png +0 -0
  20. data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00008.png +0 -0
  21. data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00009.png +0 -0
  22. data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00000.png +0 -0
  23. data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00001.png +0 -0
  24. data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00002.png +0 -0
  25. data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00003.png +0 -0
  26. data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00004.png +0 -0
  27. data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00005.png +0 -0
  28. data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00006.png +0 -0
  29. data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00007.png +0 -0
  30. data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00008.png +0 -0
  31. data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00009.png +0 -0
  32. data/gt_mask/-1HyqEM_VwM_137000_147000/fid_2/00002.png +0 -0
  33. data/gt_mask/-1HyqEM_VwM_137000_147000/fid_2/00003.png +0 -0
  34. data/gt_mask/-1HyqEM_VwM_137000_147000/fid_2/00005.png +0 -0
  35. data/gt_mask/-1HyqEM_VwM_137000_147000/fid_2/00006.png +0 -0
configs/config.py CHANGED
@@ -74,6 +74,20 @@ parser.add_argument("--frame_n", type=int, default=10, help="Frame num of each v
74
  parser.add_argument("--text_max_len", type=int, default=25, help="Maximum textual reference length.")
75
  parser.add_argument("--max_eval_rows", type=int, default=-1, help="Max samples per split during eval; -1 = all.")
76
  parser.add_argument("--eval_split", type=str, default="test_u", help="Which split to evaluate: test_s, test_u, test_n.")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
77
 
78
 
79
 
 
74
  parser.add_argument("--text_max_len", type=int, default=25, help="Maximum textual reference length.")
75
  parser.add_argument("--max_eval_rows", type=int, default=-1, help="Max samples per split during eval; -1 = all.")
76
  parser.add_argument("--eval_split", type=str, default="test_u", help="Which split to evaluate: test_s, test_u, test_n.")
77
+ parser.add_argument("--gate_only", action="store_true", help="Train only A-min referent gate parameters.")
78
+ parser.add_argument("--init_from_saved_model", action="store_true", help="Initialize training from --saved_model before updates.")
79
+ parser.add_argument("--max_steps", type=int, default=-1, help="Max optimizer steps during training; -1 = full schedule.")
80
+ parser.add_argument("--overfit_samples", type=int, default=-1, help="Train on the first N train samples for overfit probes; -1 = full train set.")
81
+ parser.add_argument("--log_gate_stats_every", type=int, default=-1, help="Log A-min gate/proj stats every N optimizer steps; -1 = disabled.")
82
+ parser.add_argument("--skip_eval_after_train", action="store_true", help="Save checkpoint and exit without post-train evaluation.")
83
+ parser.add_argument("--eval_train_only", action="store_true", help="After training, evaluate only the training subset and skip test splits.")
84
+ parser.add_argument("--cache_root", type=str, default="/workspace/SimToken/cache_q", help="Root directory for cached q features.")
85
+ parser.add_argument("--cache_split", type=str, default="train", help="Dataset split to cache or read cached q features from.")
86
+ parser.add_argument("--overwrite_cache", action="store_true", help="Overwrite existing cached q feature files.")
87
+ parser.add_argument("--eval_only", action="store_true", help="Only evaluate in cached-gate scripts; do not train.")
88
+ parser.add_argument("--disable_gate", action="store_true", help="Force A-min gate to identity for cached pipeline baseline checks.")
89
+ parser.add_argument("--gate_checkpoint", type=str, default="", help="Optional referent-gate-only checkpoint to overlay after loading --saved_model.")
90
+ parser.add_argument("--save_gate_only", action="store_true", help="In cached-gate training, save only referent_gate parameters.")
91
 
92
 
93
 
data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00000.png ADDED
data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00001.png ADDED
data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00002.png ADDED
data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00003.png ADDED
data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00004.png ADDED
data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00005.png ADDED
data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00006.png ADDED
data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00007.png ADDED
data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00008.png ADDED
data/gt_mask/--KCIeTv6PM_14000_24000/fid_0/00009.png ADDED
data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00000.png ADDED
data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00001.png ADDED
data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00002.png ADDED
data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00003.png ADDED
data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00004.png ADDED
data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00005.png ADDED
data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00006.png ADDED
data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00007.png ADDED
data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00008.png ADDED
data/gt_mask/--iSerV5DbY_68000_78000/fid_1/00009.png ADDED
data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00000.png ADDED
data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00001.png ADDED
data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00002.png ADDED
data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00003.png ADDED
data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00004.png ADDED
data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00005.png ADDED
data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00006.png ADDED
data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00007.png ADDED
data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00008.png ADDED
data/gt_mask/-1HyqEM_VwM_137000_147000/fid_0/00009.png ADDED
data/gt_mask/-1HyqEM_VwM_137000_147000/fid_2/00002.png ADDED
data/gt_mask/-1HyqEM_VwM_137000_147000/fid_2/00003.png ADDED
data/gt_mask/-1HyqEM_VwM_137000_147000/fid_2/00005.png ADDED
data/gt_mask/-1HyqEM_VwM_137000_147000/fid_2/00006.png ADDED