From 0af84388b8ec99037af30f9de272c6288850e8ec Mon Sep 17 00:00:00 2001
From: BobYeah <635596704@qq.com>
Date: Fri, 25 Dec 2020 18:47:26 +0800
Subject: [PATCH] fix a bug in run_lf_syn.py

---
 run_lf_syn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/run_lf_syn.py b/run_lf_syn.py
index b221d86..5289638 100644
--- a/run_lf_syn.py
+++ b/run_lf_syn.py
@@ -58,7 +58,7 @@ def train():
     # 3. Train
     model.train()
     epoch = EPOCH_BEGIN
-    iters = EPOCH_BEGIN * len(train_data_loader)
+    iters = EPOCH_BEGIN * len(train_data_loader) * BATCH_SIZE
 
     util.CreateDirIfNeed(RUN_DIR)
 
-- 
GitLab