[hotfix] swintrainyaml

boostcampaitech3 · May 4, 2022 · 1bba5ea · 1bba5ea
1 parent a9e2ffc
commit 1bba5ea
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/Swin-Transformer-Semantic-Segmentation/simple_train.py b/Swin-Transformer-Semantic-Segmentation/simple_train.py
@@ -48,11 +48,11 @@ def main(config_train):
 		cfg.checkpoint_config['interval'] = config_train['eval_interval']
 
 	# -- dataset
-	if config_train['train_data_dir_root']:
-		cfg.data['train']['img_dir']=os.path.join(config_train['train_data_dir_root'],'images/training')
-		cfg.data['train']['ann_dir']=os.path.join(config_train['train_data_dir_root'],'annotations/training')
-		cfg.data['val']['img_dir']=os.path.join(config_train['train_data_dir_root'],'images/training')
-		cfg.data['val']['ann_dir']=os.path.join(config_train['train_data_dir_root'],'annotations/training')
+	if config_train['data_dir_root']:
+		cfg.data['train']['img_dir']=os.path.join(config_train['data_dir_root'],'images/training')
+		cfg.data['train']['ann_dir']=os.path.join(config_train['data_dir_root'],'annotations/training')
+		cfg.data['val']['img_dir']=os.path.join(config_train['data_dir_root'],'images/validation')
+		cfg.data['val']['ann_dir']=os.path.join(config_train['data_dir_root'],'annotations/validation')
 	datasets = [build_dataset(cfg.data.train)]
 
 	# -- build model

diff --git a/Swin-Transformer-Semantic-Segmentation/train.yaml b/Swin-Transformer-Semantic-Segmentation/train.yaml
@@ -3,13 +3,13 @@ config_dir : '_base_/custom'
 config_file : 'custom_config'
 
 ## -- wandb
-wandb : False
+wandb : True
 wandb_proj : 'Semantic Segmentation'
 wandb_name : 'uperswin_512_512'
 log_interval : 50
 
 # -- hyperparameter
-seed : 1995
+seed : 84
 batch_size : 8
 lr : 0.00006
 
@@ -24,5 +24,5 @@ ckpt_name : 'upernet_swin_base_patch4_window7_512x512'
 ## -- save best model
 save_best_model : True
 
-train_data_dir_root : '/opt/ml/input/mmseg'
+data_dir_root : '/opt/ml/input/mmseg'