AFTer-UNet Architecture on SegTHOR Dataset
Training the dataset:
python3 train.py config/train_config.yaml config/test_config.yaml --dataset_to_use segthor_data
python3 train.py config/train_config.yaml config/test_config.yaml --dataset_to_use brats_data