MultivexAI
/

RobustMNIST-v1.0

Image Classification

computer-vision

Eval Results (legacy)

Model card Files Files and versions

MultivexAI commited on 6 days ago

Commit

1815e62

·

verified ·

1 Parent(s): 478c30b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -97,7 +97,7 @@ def execute():
     processor = HierarchicalNetwork(out_dims=11).to(HARDWARE_TARGET)
     # load state parameters
-    state_data = torch.load(PARAMETER_PATH, map_location=HARDWARE_TARGET)
     weights = state_data.get('state_dict', state_data)
     processor.load_state_dict(weights)
     processor.eval()

     processor = HierarchicalNetwork(out_dims=11).to(HARDWARE_TARGET)
     # load state parameters
+    state_data = torch.load(PARAMETER_PATH, map_location=HARDWARE_TARGET, weights_only=True)
     weights = state_data.get('state_dict', state_data)
     processor.load_state_dict(weights)
     processor.eval()