FT start iter update

2019-12-27 11:01:01 +00:00 · 2019-12-27 11:01:01 +00:00 · 307e776a11
parent 6a55be4d91
commit 307e776a11
1 changed files with 5 additions and 2 deletions
--- a/train.py
+++ b/train.py
@ -123,8 +123,11 @@ def train(opt):
    """ start training """
    start_iter = 0
    if opt.saved_model != '':
+        try:
            start_iter = int(opt.saved_model.split('_')[-1].split('.')[0])
            print(f'continue to train, start_iter: {start_iter}')
+        except:
+            pass

    start_time = time.time()
    best_accuracy = -1