0
点赞
收藏
分享

微信扫一扫

pytorch-mmsegmentation train时遇到 ‘torch.distributed‘ has no attribute ‘group‘

爱动漫建模 2022-01-30 阅读 139

直接搜出来说是因为torch版本不匹配,但是指定版本时怎么改呢,主要是由于模型代码中使用到了SyncBN而该方法是分布式训练时才可以调用的,在单卡GPU训练时改成BN即可
。(具体修改位置要看自己的报错,在模型哪个位置使用到了SyncVBN)请添加图片描述![请添加图片描述](https://img-blog.csdnimg.cn/b0671c39df0946cf81eff944f8dbc9aa.png在这里插入图片描述这位博主遇到的问题和我也很相似**【AssertionError:Default process group is not initialized】**pytorch-mmsegmentation train时遇到AssertionError:Default process group is not initialized,解决方式一样。

举报

相关推荐

0 条评论