source /opt/dtk-24.04.1/env.sh
export HIP_VISIBLE_DEVICES=6,7
export HSA_FORCE_FINE_GRAIN_PCIE=1

#export HIP_LOG_LEVEL=1 
#export HIP_MODULE_MASK=0x7FFFFFFF 
#export HIP_ENABLE_LIST="hip"
#export MIOPEN_ENABLE_LOGGING_CMD=1
#export MIOPEN_LOG_LEVEL=5
#export MIOPEN_DEBUG_HIPNN_CONV_S868=0
export MIOPEN_DEBUG_HIPNN_CONV_S390=0
#export MIOPEN_PRECISION_FP32_FP32_FP32_TF32_FP32=1
python train.py -c configs/ljs_base.json -m ljs_base