‘DistributedDataParallel‘ object has no attribute ‘generate‘(gpu分布式训练提示找不到model中的函数)

在使用DistributedDataParallel训练model的时候,发现在进行forward的过程中,会碰到DistributedDataParallel' object has no attribute的问题。观察可以看到此时的model已经被封装进类似DistributedDataParallel的类里,此model非平常使用的model,所以导致调用原始model的函数时会出现object has no attribute “xxxx”的bug。

解决方法:

        model = model.module即可

你可能感兴趣的:(深度学习)