fix: no need for retain_graph to be True on losses

jolibrain · Oct 18, 2022 · b715638 · b715638
1 parent d6c3a5d
commit b715638
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/base_model.py b/models/base_model.py
@@ -1025,7 +1025,7 @@ def optimize_parameters(self):
                     )
                 else:
                     ll = getattr(self, loss) / self.opt.train_iter_size
-                ll.backward(retain_graph=True)
+                ll.backward(retain_graph=False)
 
             loss_names = []