Fix model.eval()

Author: synml

demo.py

@@ -20,6 +20,7 @@
     # 2. Model
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     model = utils.get_model(config['model_name'], config['num_classes'], config['pretrained_weights']).to(device)
+    model.eval()
 
     # 이미지 이름 불러오기
     image_names = []

eval.py

@@ -96,6 +96,7 @@ def evaluate(model, testloader, num_classes: int, device):
     # 2. Model
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     model = utils.get_model(config['model_name'], config['num_classes'], config['pretrained_weights']).to(device)
+    model.eval()
 
     # 모델 평가
     val_loss, iou, miou, fps = evaluate(model, testloader, config['num_classes'], device)

feature_visualizer.py

@@ -26,6 +26,7 @@ def hook(model, input, output):
     # 2. Model
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     model = utils.get_model(config['model_name'], config['num_classes'], config['pretrained_weights']).to(device)
+    model.eval()
 
     # 이미지 불러오기
     image_number = input('Enter the image number of the dataset>>> ')