Densenet121-OSCC / README.md
sadhvikbathini's picture
Update README.md
eeb8bde verified
metadata
license: gpl-3.0

Usage:

model = torchvision.models.densenet121(pretrained=False)

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