|
@@ -384,7 +384,7 @@ class BaseModel:
|
|
|
|
|
|
|
|
# 每间隔save_interval_epochs, 在验证集上评估和对模型进行保存
|
|
# 每间隔save_interval_epochs, 在验证集上评估和对模型进行保存
|
|
|
if ema is not None:
|
|
if ema is not None:
|
|
|
- weight = self.net.state_dict()
|
|
|
|
|
|
|
+ weight = copy.deepcopy(self.net.state_dict())
|
|
|
self.net.set_state_dict(ema.apply())
|
|
self.net.set_state_dict(ema.apply())
|
|
|
eval_epoch_tic = time.time()
|
|
eval_epoch_tic = time.time()
|
|
|
if (i + 1) % save_interval_epochs == 0 or i == num_epochs - 1:
|
|
if (i + 1) % save_interval_epochs == 0 or i == num_epochs - 1:
|