Commit 0af84388 authored by BobYeah's avatar BobYeah
Browse files

fix a bug in run_lf_syn.py

parent 67c4de9e
...@@ -58,7 +58,7 @@ def train(): ...@@ -58,7 +58,7 @@ def train():
# 3. Train # 3. Train
model.train() model.train()
epoch = EPOCH_BEGIN epoch = EPOCH_BEGIN
iters = EPOCH_BEGIN * len(train_data_loader) iters = EPOCH_BEGIN * len(train_data_loader) * BATCH_SIZE
util.CreateDirIfNeed(RUN_DIR) util.CreateDirIfNeed(RUN_DIR)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment