rm -rf ./exp
export HIP_MODULE_LOADING=EAGER
export HIP_ALLOC_INITIALIZE=0
export HIP_REUSE_MODULE=1
export HIP_VISIBLE_DEVICES=0
config_file=./config/AFNO-50epoch.yaml
config='afno_backbone'
run_num='check_exp'
export HDF5_USE_FILE_LOCKING=FALSE
python train.py --enable_amp --yaml_config=$config_file --config=$config --run_num=$run_num