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