From 27048af49958497610a82e8614b8dbab78f9698f Mon Sep 17 00:00:00 2001 From: Coby Mulliken Date: Thu, 31 Jul 2025 14:12:32 -0400 Subject: [PATCH] strict=True --- brainscore_vision/models/chresmax_v3_cl_01/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/brainscore_vision/models/chresmax_v3_cl_01/model.py b/brainscore_vision/models/chresmax_v3_cl_01/model.py index dc561036a..29f90f04a 100755 --- a/brainscore_vision/models/chresmax_v3_cl_01/model.py +++ b/brainscore_vision/models/chresmax_v3_cl_01/model.py @@ -996,7 +996,7 @@ def load_model(model_name, bypass=True, main_route=False, c_scoring='v2' ).to(device).eval() - model.load_state_dict(checkpoint['state_dict'], strict=False) + model.load_state_dict(checkpoint['state_dict'], strict=True) return model else: raise ValueError(f"Unsupported model: {model_name}")