export HIP_VISIBLE_DEVICES=0,1,2,3
python -m torch.distributed.launch --master_port 50130 --nnodes 1  --nproc_per_node=4 --use_env  main.py \
 --model Conformer_small_patch16 \
 --data-set IMNET \
 --batch-size 64 \
 --lr 0.001 \
 --num_workers 4 \
 --data-path /public/software/apps/DeepLearning/Data/ImageNet-pytorch \
 --output_dir /public/home/hepj/SothisAI/Conformer-main/test_dir  \
 --epochs 1
