刘钊
刘钊
fit_one_epoch 的前两个参数model_train 和model 在train.py 中 model_train = model.train() 我理解为此时model和model_train是同一个东西 但是如果使用多卡，model_train = torch.nn.DataParallel(model) 这时候他们就不一样了吧 训练过程中更新的是model_train的参数，但是最后却保存了model的。不知这样做有什么目的