diff options
author | Oleksii Kachaiev <kachayev@gmail.com> | 2023-05-02 08:42:32 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-05-02 08:42:32 +0200 |
commit | 8a7035bdaa5bb164d1c16febbd83650d1fb6d393 (patch) | |
tree | e2cd81c8b5d0c6d78f9bce603135be5783138785 | |
parent | 42a62c123776e04ee805aefb9afd6d98abdcf192 (diff) |
Perform zero_grad after each batch (#466)
-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 |