cd /home/v-qichen3/MRI_recon/code/Frequency-Diffusion/FSMNet

data_root=/home/v-qichen3/MRI_recon/data/fastmri


python train_fastmri.py --root_path $data_root \
    --gpu 2 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.08 --ACCELERATIONS 4 \
    --exp FSMNet_fastmri_4x  --MRIDOWN 4X  --MASKTYPE random   \
    --num_timesteps 5 --image_size 320 --use_kspace  --use_time_model 

# fastMRI dataset, AF=8

# python train_fastmri.py --root_path $data_root  \
#     --gpu 0 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.04 --ACCELERATIONS 8 \
#     --exp FSMNet_fastmri_8x  --MRIDOWN 8X --MASKTYPE equispaced  \
#     --num_timesteps 5 --image_size 320 --use_kspace  --use_time_model 


# python train_fastmri.py --root_path $data_root  \
#     --gpu 0 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.03 --ACCELERATIONS 12 \
#     --exp FSMNet_fastmri_12x  --MRIDOWN 12X --MASKTYPE equispaced   \
#     --num_timesteps 5 --image_size 320 --use_kspace  --use_time_model 



# # Test
# #fastMRI dataset, AF=4
# model_4x=model/FSMNet_fastmri_4x/iter_100000.pth

python test_fastmri.py --root_path $data_root \
    --gpu 1 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.08 --ACCELERATIONS 4 \
    --exp FSMNet_fastmri_4x --phase test   --MRIDOWN 4X   \
    --num_timesteps 5 --image_size 320 --use_kspace  --use_time_model  --test_sample Ksample --snapshot_path /home/v-qichen3/MRI_recon/code/Frequency-Diffusion/FSMNet/model/FSMNet_fastmri_4x_t5_kspace_time_no_distortion
 
 
 
 # ColdDiffusion  DDPM

# #fastMRI dataset, AF=8

python test_fastmri.py --root_path $data_root \
    --gpu 3 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.04 --ACCELERATIONS 8 \
    --exp FSMNet_fastmri_8x --phase test  --MRIDOWN 8X   \
    --num_timesteps 5 --image_size 320 --use_kspace  --use_time_model --test_sample Ksample --snapshot_path /home/v-qichen3/MRI_recon/code/Frequency-Diffusion/FSMNet/model/FSMNet_fastmri_8x_t5_kspace_time_no_distortion


 # ColdDiffusion  DDPM


# python test_fastmri.py --root_path $data_root \
#     --gpu 0 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.03 --ACCELERATIONS 12 \
#     --exp FSMNet_fastmri_12x --phase test  --MRIDOWN 12X   \
#     --num_timesteps 5 --image_size 320 --use_kspace  --use_time_model        --test_sample Ksample  # ColdDiffusion  DDPM



# # FSMNet_fastmri_12x_t30_kspace_time