File size: 146 Bytes
0164e2e
 
 
eeb8bde
0164e2e
eeb8bde
 
0164e2e
1
2
3
4
5
6
7
8
9
---
license: gpl-3.0
---
Usage:
---
model = torchvision.models.densenet121(pretrained=False)

model.load_state_dict(torch.load('best_model.pth'))