alcheung0213
/

Medmnist_active_learning

Image Classification

active-learning

medical-imaging

Model card Files Files and versions

alcheung0213 commited on Nov 21, 2024

Commit

baa7e0f

·

1 Parent(s): cc7a0e7

fixing errors in readme

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -62,7 +62,7 @@ To utilize this model:
     from model import ResNet50_28
     model = ResNet50_28(num_classes=9)
-    model.load_state_dict(torch.load('path_to_checkpoint.pt')['model_state_dict'])
     model.eval()
     ```

     from model import ResNet50_28
     model = ResNet50_28(num_classes=9)
+    model.load_state_dict(torch.load('pytorch_model.bin'))
     model.eval()
     ```