solution-train-two-model-simultaneously
同时训练两个模型遇到了bug,记录今天由这个bug引发一系列bug的过程
使用resnet18进行图像分类
1234567model_ft = models.resnet18(pretrained=True)num_ftrs = model_ft.fc.in_featuresmodel_ft.fc = nn.Linear(num_ftrs, len(class_names))model_ft = model_ft.to(device)cross_entropy_loss = nn.CrossEntropyLoss()optimizer_ft = optim.SGD(model_ft.parameters(), lr=1e-3, momentum=0.9)exp_lr_scheduler = lr_scheduler.StepLR(optimizer_ft, step_size=5, gamma=0.1)
单独训练这个模型在我的数据集的正确率在0.96左右
需求:同时跑两个一样模型
于是我定义了一个函数,使用deepcopy复制得到全新的模型,并且在train函数中同时跑这两个模型
12 ...