Densenet121-OSCC / README.md
sadhvikbathini's picture
Update README.md
eeb8bde verified
---
license: gpl-3.0
---
Usage:
---
model = torchvision.models.densenet121(pretrained=False)
model.load_state_dict(torch.load('best_model.pth'))