remove opt step from test

micromind-toolkit · Nov 15, 2023 · 9fb50f3 · 9fb50f3
1 parent 1008077
commit 9fb50f3
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/micromind/core.py b/micromind/core.py
@@ -529,7 +529,6 @@ def test(self, datasets: Dict = {}) -> None:
             for idx, batch in enumerate(pbar):
                 if isinstance(batch, list):
                     batch = [b.to(self.device) for b in batch]
-                self.opt.zero_grad()
 
                 model_out = self(batch)
                 loss = self.compute_loss(model_out, batch)