Hugging Face's logo

SoulPerforms
/
Butterfly_image_classification_resnet18

Image Classification
Transformers
PyTorch
English
resnet
biology
Model card Files
xet
 Community
1
SoulPerforms commited on
Commit
ccb41e8
·
verified ·
1 Parent(s): 2c3fd62

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -26,7 +26,7 @@ Epoch: 28 Train Loss: 0.17 Train Accuracy: 0.96 Test Accuracy: 0.90
26
  2. load pretrained model resnet18
27
  3. model_for_predict = models.resnet18(pretrained=True)
28
  4. load checkpoint from your local
29
- 5. checkpoint = torch.load('butterfly_resnet_checkpoint.model')
30
  7. model_for_predict.load_state_dict(checkpoint)
31
  8. predict the images
32
  9. model_for_predict.eval())
 
26
  2. load pretrained model resnet18
27
  3. model_for_predict = models.resnet18(pretrained=True)
28
  4. load checkpoint from your local
29
+ 5. checkpoint = torch.load('pytorch_model.bin')
30
  7. model_for_predict.load_state_dict(checkpoint)
31
  8. predict the images
32
  9. model_for_predict.eval())