‘DistributedDataParallel‘ object has no attribute ‘generate‘(gpu分布式训练提示找不到model中的函数)
在使用DistributedDataParallel训练model的时候，发现在进行forward的过程中，会碰到DistributedDataParallel' object has no attribute的问题。观察可以看到此时的model已经被封装进类似DistributedDataParallel的类里，此model非平常使用的model，所以导致调用原始model的函数时会出现obje
·
在使用DistributedDataParallel训练model的时候，发现在进行forward的过程中，会碰到DistributedDataParallel' object has no attribute的问题。观察可以看到此时的model已经被封装进类似DistributedDataParallel的类里，此model非平常使用的model，所以导致调用原始model的函数时会出现object has no attribute “xxxx”的bug。
解决方法：
model = model.module即可
更多推荐
4
0- 0
已为社区贡献1条内容
所有评论(0)