fix deepcopy bug

tim-learn · web-flow · commit f7d555a0d53b · 2024-02-22T16:44:46.000+08:00
diff --git a/digit/uda_digit.py b/digit/uda_digit.py
@@ -208,9 +208,9 @@ def train_source(args):
 
             if acc_s_te >= acc_init:
                 acc_init = acc_s_te
-                best_netF = netF.state_dict()
-                best_netB = netB.state_dict()
-                best_netC = netC.state_dict()
+                best_netF = copy.deepcopy(netF.state_dict())
+                best_netB = copy.deepcopy(netB.state_dict())
+                best_netC = copy.deepcopy(netC.state_dict())
             
             netF.train()
             netB.train()
