diff options
-rw-r--r-- | examples/backends/plot_wass2_gan_torch.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/examples/backends/plot_wass2_gan_torch.py b/examples/backends/plot_wass2_gan_torch.py index ca5b3c9..cc82f4f 100644 --- a/examples/backends/plot_wass2_gan_torch.py +++ b/examples/backends/plot_wass2_gan_torch.py @@ -151,6 +151,7 @@ for i in range(n_iter): loss.backward() optimizer.step() + optimizer.zero_grad() del M |