train: Fix saving best val loss weights

Author: synml

train.py

@@ -34,7 +34,7 @@
     # 5. Train and evaluate
     log_loss = tqdm.tqdm(total=0, position=2, bar_format='{desc}', leave=False)
     prev_miou = 0.0
-    prev_val_loss = 0.0
+    prev_val_loss = 100
     for epoch in tqdm.tqdm(range(config[config['model']]['epoch']), desc='Epoch'):
         if utils.train_interupter():
             print('Train interrupt occurs.')

train_interupter.txt renamed to train_interupter.ini

@@ -194,13 +194,13 @@ def __call__(self, image, target):
 
 
 def train_interupter():
-    with open('train_interupter.txt', 'r', encoding='utf-8') as f:
+    with open('train_interupter.ini', 'r', encoding='utf-8') as f:
         flag = f.read().strip()
 
     if flag == '0':
         return False
     elif flag == '1':
-        with open('train_interupter.txt', 'w', encoding='utf-8') as f:
+        with open('train_interupter.ini', 'w', encoding='utf-8') as f:
             f.write('0')
         return True
     else: