export HIP_VISIBLE_DEVICES=1
export MIOPEN_FIND_MODE=1
my_config=$1
python3 tools/train.py $my_config --gpu-ids 0