CELL 15 — RUN SCREENING SUITE FOR BOTH CHECKPOINTS

screen_records = []
all_results = {}

def run_screen_for_checkpoint(epoch, adapter):
model = load_checkpoint(adapter)

text
try:
    for domain, df in [
        ("clean", clean_screen_df),
        ("phone", robust_val_df),
    ]:
        for view_mode in [
            "FULL_PAGE",
            "MOSAIC",
        ]:
            key = f"E{epoch}_{domain}_{view_mode}"

            print("\n" + "=" * 80)
            print(key)
            print("=" * 80)

            result = evaluate_view(
                model,
                df,
                domain=domain,
                view_mode=view_mode,
                desc=key,
            )

            all_results[key] = result

            save_result(
                result,
                OUTPUT_DIR / "screening" / key,
            )

            screen_records.append({
                "epoch": epoch,
                "domain": domain,
                "view_mode": view_mode,
                "n": len(df),
                "accuracy": result["accuracy"],
                "macro_f1": result["macro_f1"],
                "balanced_accuracy": result["balanced_accuracy"],
            })

            print(
                f"Accuracy={result['accuracy']:.4f} | "
                f"Macro-F1={result['macro_f1']:.4f} | "
                f"BalAcc={result['balanced_accuracy']:.4f}"
            )
finally:
    release_model(model)

run_screen_for_checkpoint(6, EPOCH6_ADAPTER)
run_screen_for_checkpoint(7, EPOCH7_ADAPTER)

screen_df = pd.DataFrame(screen_records)
screen_df.to_csv(
OUTPUT_DIR / "SCREENING_RESULTS.csv",
index=False,
)

display(screen_df)

E6_clean_FULL_PAGE

Accuracy=0.4286 | Macro-F1=0.3924 | BalAcc=0.4162

================================================================================
E6_clean_MOSAIC

---

FloatingPointError Traceback (most recent call last)
/tmp/ipykernel_23/3349340959.py in evaluate_view(eval_model, df, domain, view_mode, desc)
47 with precision_context():
---> 48 logits = constrained_class_logits(
49 eval_model,

/tmp/ipykernel_23/4098905817.py in constrained_class_logits(eval_model, batch, diagnose_nonfinite, force_vision_fp32)
177
--> 178 hidden_states = qwen_hidden_4513(
179 causal_lm,

/tmp/ipykernel_23/4098905817.py in qwen_hidden_4513(causal_lm, model_batch, diagnose_nonfinite, force_vision_fp32)
81 if diagnose_nonfinite and not torch.isfinite(image_embeds).all():
---> 82 raise_stage_nonfinite("vision", image_embeds)
83

/tmp/ipykernel_23/4098905817.py in raise_stage_nonfinite(stage, tensor)
30 frac = float(torch.isfinite(tensor).float().mean().detach().cpu())
---> 31 raise FloatingPointError(
32 f"NONFINITE_STAGE={stage} dtype={tensor.dtype} "

FloatingPointError: NONFINITE_STAGE=vision dtype=torch.float16 shape=(475, 2048) finite_fraction=0.99789470

The above exception was the direct cause of the following exception:

RuntimeError Traceback (most recent call last)
/tmp/ipykernel_23/2691407900.py in <cell line: 0>()
55 release_model(model)
56
---> 57 run_screen_for_checkpoint(6, EPOCH6_ADAPTER)
58 run_screen_for_checkpoint(7, EPOCH7_ADAPTER)
59

/tmp/ipykernel_23/2691407900.py in run_screen_for_checkpoint(epoch, adapter)
22 print("=" * 80)
23
---> 24 result = evaluate_view(
25 model,
26 df,

/usr/local/lib/python3.13/dist-packages/torch/utils/_contextlib.py in decorate_context(*args, **kwargs)
122 # pyrefly: ignore [bad-context-manager]
123 with ctx_factory():
--> 124 return func(*args, **kwargs)
125
126 return decorate_context

/tmp/ipykernel_23/3349340959.py in evaluate_view(eval_model, df, domain, view_mode, desc)
53 )
54 except Exception as exc:
---> 55 raise RuntimeError(
56 f"Evaluation failed: domain={domain}, "
57 f"view={view_mode}, "

RuntimeError: Evaluation failed: domain=clean, view=MOSAIC, sample=4963.0|2297|bw_clean, label=CD, aug=clean

Can you give the fixed cell or whole notebook fixed please