bug fix

CASIA-IVA-Lab · Sep 14, 2023 · 72642ff · 72642ff
1 parent fb0b086
commit 72642ff
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/code/train_mvtec.py b/code/train_mvtec.py
@@ -85,13 +85,12 @@ def main(**args):
         iter_every_epoch = 0
         for batch, batch_sft in zip(train_iter,train_iter_sft):
             iter_every_epoch += 1
-            if iter_every_epoch % 2 == 0:
-                agent.train_model(
-                    batch, 
-                    current_step=current_step, 
-                    pbar=pbar
-                )
-                del batch
+            agent.train_model(
+                batch, 
+                current_step=current_step, 
+                pbar=pbar
+            )
+            del batch
 
             agent.train_model(
                 batch_sft,