From ca00f4b63b3df78eb2eb32727a610569471860a5 Mon Sep 17 00:00:00 2001
From: JiaQi Xu <47347516+bubbliiiing@users.noreply.github.com>
Date: Sun, 29 Mar 2020 14:33:15 +0800
Subject: [PATCH] Update train.py

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

diff --git a/train.py b/train.py
index 0cf898e..2efd283 100644
--- a/train.py
+++ b/train.py
@@ -60,7 +60,7 @@ if __name__ == "__main__":
 
 
     epoch_size = num_train // Batch_size
-    for epoch in range(Epoch):
+    for epoch in range(Start_iter,Epoch):
         if epoch%10==0:
             adjust_learning_rate(optimizer,lr,0.95,epoch)
         loc_loss = 0
-- 
GitLab