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

[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