# BraTS dataset, AF=4
mamba activate diffmri
cd /home/cbtil3/hao/repo/Frequency-Diffusion/experiments/FSMNet

#root_path=/bask/projects/j/jiaoj-rep-learn/Hao/datasets/knee/image_100patients_4X/
root_path=/gamedrive/Datasets/medical/FrequencyDiffusion/brats/image_100patients_4X/

python train_brats.py --root_path $root_path\
    --gpu 0 --batch_size 4 --base_lr 0.0001 --MRIDOWN 4X --low_field_SNR 0 \
    --input_normalize mean_std \
    --exp FSMNet_BraTS_4x

BraTS dataset, AF=8

python train_brats.py --root_path /gamedrive/Datasets/medical/FrequencyDiffusion/brats/image_100patients_8X/ \
    --gpu 1 --batch_size 4 --base_lr 0.0001 --MRIDOWN 8X --low_field_SNR 0 \
    --input_normalize mean_std \
    --exp FSMNet_BraTS_8x




# Test
BraTS dataset, AF=4

python test_brats.py --root_path /data/qic99/MRI_recon/image_100patients_4X/ \
    --gpu 3 --base_lr 0.0001 --MRIDOWN 4X --low_field_SNR 0 \
    --input_normalize mean_std \
    --exp FSMNet_BraTS_4x --phase test

BraTS dataset, AF=8

python test_brats.py --root_path /data/qic99/MRI_recon/image_100patients_8X/ \
    --gpu 4 --base_lr 0.0001 --MRIDOWN 8X --low_field_SNR 0 \
    --input_normalize mean_std \
    --exp FSMNet_BraTS_8x --phase test