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
