分布式训练卡住了 如下图所示

[vousemevoyez1024]

[2022-08-31 16:56:10,107-rk0-model_load.py# 42] remove prefix 'module.' [2022-08-31 16:56:10,108-rk0-model_load.py# 33] used keys:265 [2022-08-31 16:56:10,116-rk0-train.py# 58] build train dataset [2022-08-31 16:56:10,117-rk0-dataset.py# 39] loading VID [2022-08-31 16:56:10,117-rk0-dataset.py# 64] VID loaded

命令为： CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch \ --nproc_per_node=2 \ --master_port=2333 \ ../../tools/train.py --cfg config.yaml