v0.4.9: vision-aware pooling (RCA fix) + v4 namespace + eps0.45 collect config
Browse files- 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 |
)
|