春笋

i use torchvision.models for model zoo [code] def Create_Model_From_Zoo(num_classes=4): model = models.resnet50(pretrained=True) num_ftrs = model.fc.in_features model.fc = nn.Linear(num_ftrs,num_classes) classy_model = ClassyModel.from_model(model) return classy_model ... config_file='./configs/test.json' config = load_json(config_file) model =...