Update otx/algorithms/segmentation/adapters/mmseg/tasks/trainer.py

Co-authored-by: Eunwoo Shin <eunwoo.shin@intel.com>
openvinotoolkit · Mar 24, 2023 · de6ec1c · de6ec1c
1 parent 9085a92
commit de6ec1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/otx/algorithms/segmentation/adapters/mmseg/tasks/trainer.py b/otx/algorithms/segmentation/adapters/mmseg/tasks/trainer.py
@@ -99,7 +99,7 @@ def run(self, model_cfg, model_ckpt, data_cfg, **kwargs):
         # cfg.dump(osp.join(cfg.work_dir, 'config.py'))
         # logger.info(f'Config:\n{cfg.pretty_text}')
 
-        validate = cfg.data.get("val", None)
+        validate = "val" in cfg.data
         train_segmentor(
             model,
             datasets,