update scripts

Author: ajayvohra2005

Dockerfile.neuron

@@ -1,4 +1,4 @@
-FROM public.ecr.aws/neuron/pytorch-training-neuronx:2.5.1-neuronx-py310-sdk2.21.0-ubuntu22.04
+FROM public.ecr.aws/neuron/pytorch-training-neuronx:2.5.1-neuronx-py310-sdk2.22.0-ubuntu22.04
 
 ENV DEBIAN_FRONTEND=noninteractive
 ENV DEBCONF_NONINTERACTIVE_SEEN=true

run-neuron.sh

@@ -2,11 +2,11 @@ export YOLOX_DATADIR=/datasets
 export YOLOX_OUPUT_DIR="./YOLOX_neuron_outputs"
 export OMP_NUM_THREADS=16
 export LOGURU_LEVEL="INFO"
-export NEURON_CC_FLAGS="--cache_dir=/cache --model-type=cnn-training"
+export NEURON_CC_FLAGS="--cache_dir=/cache --model-type=generic"
 export NEURON_RT_STOCHASTIC_ROUNDING_EN="1"
 export XLA_IR_SHAPE_CACHE_SIZE="20480"
-export XLA_IR_DEBUG=1
-export XLA_HLO_DEBUG=1
-export PT_XLA_DEBUG=1
+export XLA_IR_DEBUG=0
+export XLA_HLO_DEBUG=0
+export PT_XLA_DEBUG=0
 export PT_XLA_DEBUG_FILE=./pt_xla_debug.txt
 torchrun --standalone --nproc_per_node=32 tools/train.py -b 128 -n yolox-s