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

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


python train_fastmri.py --root_path $data_root \
    --gpu 0 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.08 --ACCELERATIONS 4 \
    --exp FSMNet_fastmri_4x

# fastMRI dataset, AF=8

python train_fastmri.py --root_path /data/qic99/MRI_recon/fastMRI/ \
    --gpu 1 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.04 --ACCELERATIONS 8 \
    --exp FSMNet_fastmri_8x


# Test
fastMRI dataset, AF=4

python test_fastmri.py --root_path /data/qic99/MRI_recon/fastMRI/ \
    --gpu 5 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.08 --ACCELERATIONS 4 \
    --exp FSMNet_fastmri_4x --phase test

fastMRI dataset, AF=8

python test_fastmri.py --root_path /data/qic99/MRI_recon/fastMRI/ \
    --gpu 6 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.04 --ACCELERATIONS 8 \
    --exp FSMNet_fastmri_8x --phase test