train: Implement train_interupter()

Author: synml

train.py

@@ -36,6 +36,9 @@
     prev_miou = 0.0
     prev_val_loss = 0.0
     for epoch in tqdm.tqdm(range(config[config['model']]['epoch']), desc='Epoch'):
+        if utils.train_interupter():
+            print('Train interrupt occurs.')
+            break
         model.train()
 
         for batch_idx, (image, target) in enumerate(tqdm.tqdm(trainloader, desc='Train', leave=False)):

train_interupter.txt

@@ -0,0 +1 @@
+0

utils.py

@@ -187,6 +187,20 @@ def __call__(self, image, target):
         return image, target
 
 
+def train_interupter():
+    with open('train_interupter.txt', 'r', encoding='utf-8') as f:
+        flag = f.read()
+
+    if flag == '0':
+        return False
+    elif flag == '1':
+        with open('train_interupter.txt', 'w', encoding='utf-8') as f:
+            f.write('0')
+        return True
+    else:
+        raise ValueError('Wrong flag value.')
+
+
 # 데이터셋 불러오는 코드 검증 (Shape: [batch, channel, height, width])
 def show_dataset(image: torch.Tensor, target: torch.Tensor):
     def make_plt_subplot(nrows: int, ncols: int, index: int, title: str, image):