fix fastfcn

Author: CharlesPikachu

ssseg/configs/mae/base_cfg.py

@@ -6,11 +6,11 @@
     'align_corners': False,
     'backend': 'nccl',
     'work_dir': 'ckpts',
-    'logfilepath': '',
-    'log_interval_iterations': 50,
     'eval_interval_epochs': 10,
     'save_interval_epochs': 1,
-    'resultsavepath': '',
+    'evaluate_results_filename': '',
+    'logger_handle_cfg': {'type': 'LocalLoggerHandle', 'logfilepath': ''},
+    'training_logging_manager_cfg': {'log_interval_iters': 50},
     'norm_cfg': {'type': 'SyncBatchNorm'},
     'act_cfg': {'type': 'ReLU', 'inplace': True},
     'backbone': {

ssseg/configs/mae/upernet_maevitbase_ade20k.py

@@ -1,4 +1,5 @@
 '''upernet_maevitbase_ade20k'''
+import os
 import copy
 from .base_cfg import SEGMENTOR_CFG
 from .._base_ import DATASET_CFG_ADE20k_512x512, DATALOADER_CFG_BS16
@@ -15,6 +16,6 @@
 # modify other segmentor configs
 SEGMENTOR_CFG['num_classes'] = 150
 SEGMENTOR_CFG['inference']['tricks']['use_probs_before_resize'] = True
-SEGMENTOR_CFG['work_dir'] = 'upernet_maevitbase_ade20k'
-SEGMENTOR_CFG['logfilepath'] = 'upernet_maevitbase_ade20k/upernet_maevitbase_ade20k.log'
-SEGMENTOR_CFG['resultsavepath'] = 'upernet_maevitbase_ade20k/upernet_maevitbase_ade20k_results.pkl'
+SEGMENTOR_CFG['work_dir'] = os.path.split(__file__)[-1].split('.')[0]
+SEGMENTOR_CFG['evaluate_results_filename'] = f"{os.path.split(__file__)[-1].split('.')[0]}.pkl"
+SEGMENTOR_CFG['logger_handle_cfg']['logfilepath'] = os.path.join(SEGMENTOR_CFG['work_dir'], f"{os.path.split(__file__)[-1].split('.')[0]}.log")

ssseg/modules/models/segmentors/fastfcn/fastfcn.py

@@ -40,6 +40,7 @@ def __init__(self, cfg, mode):
         if cfg.get('is_freeze_norm', False): self.freezenormalization()
     '''forward'''
     def forward(self, x, targets=None, **kwargs):
+        self.segmentor.module.mode = self.mode
         return self.segmentor(x, targets, **kwargs)
     '''transforminputs'''
     def transforminputs(self, x_list, selected_indices=None):