Przeglądaj źródła

Merge pull request #908 from will-jl944/develop_jf

use deepcopy in ema
FlyingQianMM 4 lat temu
rodzic
commit
55b1e53b1f
1 zmienionych plików z 1 dodań i 1 usunięć
  1. 1 1
      dygraph/paddlex/cv/models/base.py

+ 1 - 1
dygraph/paddlex/cv/models/base.py

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