关于fit_one_epoch函数

[liuzhaoo]

fit_one_epoch 的前两个参数model_train 和model

在train.py 中 model_train = model.train() 我理解为此时model和model_train是同一个东西 但是如果使用多卡，model_train = torch.nn.DataParallel(model) 这时候他们就不一样了吧

训练过程中更新的是model_train的参数，但是最后却保存了model的。不知这样做有什么目的

[bubbliiiing]

两个参数一样，一个是多卡的，一个是单模型的参数。

[bubbliiiing]

二者只是单卡和多卡的区别