tchbcb commited on
Commit
64d4b8e
·
verified ·
1 Parent(s): 7e4ea64

v0.4.9: vision-aware pooling (RCA fix) + v4 namespace + eps0.45 collect config

Browse files
Files changed (1) hide show
  1. train/calibrate_rlcd.py +1 -0
train/calibrate_rlcd.py CHANGED
@@ -88,6 +88,7 @@ def main() -> None:
88
  score_levels=int(s1_cfg.get("score_levels", 10)),
89
  leaf_batch_size=int(s1_cfg.get(
90
  "eval_leaf_batch", s1_cfg.get("leaf_batch_size", 32))),
 
91
  tiny=TinyVLMConfig(**(s1_cfg.get("tiny") or {})),
92
  use_escalate_head=bool(s1_cfg.get("use_escalate_head", True)),
93
  )
 
88
  score_levels=int(s1_cfg.get("score_levels", 10)),
89
  leaf_batch_size=int(s1_cfg.get(
90
  "eval_leaf_batch", s1_cfg.get("leaf_batch_size", 32))),
91
+ pool_vision=bool(s1_cfg.get("pool_vision", False)),
92
  tiny=TinyVLMConfig(**(s1_cfg.get("tiny") or {})),
93
  use_escalate_head=bool(s1_cfg.get("use_escalate_head", True)),
94
  )