Update test.sh

test.sh

-export CUDA_VISIBLE_DEVICES=1,2,3,4
+export HIP_VISIBLE_DEVICES=1,2,3,4
 export HSA_FORCE_FINE_GRAIN_PCIE=1  #多卡用，强制开启PCIe细粒度模式，有助于提升多卡间通信效率。
 export USE_MIOPEN_BATCHNORM=1 # 多GPU进行并行计算时的性能优化
 
@@ -16,4 +16,4 @@ export USE_MIOPEN_BATCHNORM=1 # 多GPU进行并行计算时的性能优化
 # --wpe=0.1: 训练结束时的最终lr
 torchrun --nproc_per_node=4 --nnodes=1 --node_rank=0 train.py \
   --depth=16 --bs=192 --ep=5 --fp16=1 --alng=1e-3 --wpe=0.1 \
-  --data_path=/home/VAR/dataset
\ No newline at end of file
+  --data_path=/home/VAR/dataset