source /opt/dtk-24.04.1/env.sh
export HIP_VISIBLE_DEVICES=6

export MIOPEN_DEBUG_HIPNN_CONV_S390=0
python train.py -c configs/ljs_base.json -m ljs_base