TO run Download dataset and run it with following parameters: python baseline.py --dataset minicity --batch_size 2 --dataset_path data/minicity --output_dir unet_right_cutmix_sprinkles8b20 --cutmix_sprink_prob 0.5 --beta 2.0 --model unet --epochs 300