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
