gradient acc bug

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

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)