picekl
/

FungiCLEF2024-Sample_Submission

Model card Files Files and versions Community

picekl commited on Mar 11

Commit

4cef224

•

1 Parent(s): cba4c98

fix: unexpected key

Files changed (1) hide show

script.py +1 -1

script.py CHANGED Viewed

@@ -29,7 +29,7 @@ class PytorchWorker:
             if not torch.cuda.is_available():
                 model_ckpt = torch.load(model_path, map_location=torch.device("cpu"))
             else:
-                model_ckpt = torch.load(model_path)["model"]
             model.load_state_dict(model_ckpt)

             if not torch.cuda.is_available():
                 model_ckpt = torch.load(model_path, map_location=torch.device("cpu"))
             else:
+                model_ckpt = torch.load(model_path)
             model.load_state_dict(model_ckpt)