metadata
license: gpl-3.0
Usage:
model = torchvision.models.densenet121(pretrained=False)
model.load_state_dict(torch.load('best_model.pth'))
license: gpl-3.0
model = torchvision.models.densenet121(pretrained=False)
model.load_state_dict(torch.load('best_model.pth'))