Browse Source

use deepcopy in ema

will-jl944 4 năm trước cách đây
mục cha
commit
69acc273c9
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  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: