关于torchrun.py的问题

很感谢您的代码分享。我在进行训练的时候，总是出现卡在from .collection import imread_collection_wrapper 这里，一直没解决掉这个问题，请问这是怎么回事呢？谢谢