# InternImage for Image ClassificationThis folder contains the implementation of the InternImage for image classification!-- TOC -- Install#install Data Preparation#data-preparation Evaluation#evaluat

在使用单机4个gpu时，需要添加--local_rank参数来指定每个gpu的rank，例如：

python -m torch.distributed.launch --nproc_per_node 4 --master_port 12345 main.py \
--cfg configs/internimage_b_1k_224.yaml --data-path <imagenet-path> --local_rank 0

其中，--local_rank参数的取值范围为[0, nproc_per_node-1]。在启动训练的时候，需要为每个gpu指定一个rank，确保每个gpu的计算是独立的。