From a1b464e5dc8546286bffb1dad574b17274946c30 Mon Sep 17 00:00:00 2001
From: emetheni <lenakmeth@gmail.com>
Date: Mon, 22 May 2023 15:14:55 +0200
Subject: [PATCH] gradient acc bug
---
environment.yml | 3 +--
pytorch_classifier.py | 1 -
2 files changed, 1 insertion(+), 3 deletions(-)
diff --git a/environment.yml b/environment.yml
index 2869b25..bd0b51d 100644
--- a/environment.yml
+++ b/environment.yml
@@ -85,5 +85,4 @@ dependencies:
- tzdata==2023.3
- urllib3==2.0.2
- xxhash==3.2.0
- - yarl==1.9.2
-prefix: /home/lena/anaconda3/envs/testenv
+ - yarl==1.9.2
\ No newline at end of file
diff --git a/pytorch_classifier.py b/pytorch_classifier.py
index 0fcc27a..ad32fc7 100644
--- a/pytorch_classifier.py
+++ b/pytorch_classifier.py
@@ -194,7 +194,6 @@ def train(model,
optimizer.step() # or flip them?
optimizer.zero_grad()
-
model.zero_grad()
# loss.backward()
torch.nn.utils.clip_grad_norm_(model.parameters(), 1.0)
--
GitLab