Merge pull request #62 from onepeachbiubiubiu/requires_grad_false

fix net_disc gradient bug
GaParmar · Sep 8, 2024 · 463b2d3 · 463b2d3
2 parents 5d63339 + 296fa53
commit 463b2d3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/train_cyclegan_turbo.py b/src/train_cyclegan_turbo.py
@@ -208,6 +208,7 @@ def main(args):
                 optimizer_gen.step()
                 lr_scheduler_gen.step()
                 optimizer_gen.zero_grad()
+                optimizer_disc.zero_grad()
 
                 """
                 Identity Objective