fix a bug in run_lf_syn.py

0af84388 · BobYeah · 67c4de9e · 0af84388
Commit 0af84388 authored Dec 25, 2020 by BobYeah
--- 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)