解决监督学习，深度学习报错：AttributeError: ‘xxx‘ object has no attribute ‘module‘！！！！

最新推荐文章于 2024-05-27 21:27:23 发布

今天不想Debug

最新推荐文章于 2024-05-27 21:27:23 发布

阅读量1.9k

点赞数 2

文章标签：深度学习人工智能

本文链接：https://blog.csdn.net/weixin_45325693/article/details/132196678

版权

作者分享了解决在服务器上使用nn.DataParallel或DistributedDataParallel时遇到的AttributeError问题，原因在于代码环境差异和模型封装。解决方案包括避免`.module`使用和适配模型并行技术的代码调整。

摘要由CSDN通过智能技术生成

哈喽小伙伴们大家好呀，很长时间没有更新啦，最近在研究一个问题，就是AttributeError: 'xxx' object has no attribute 'module'

今天终于是解决了，所以来记录分享一下：

我这里出现的问题是：

因为我的数据比较大，所以我在服务器上运行程序，但是我在本机上调通了，到服务器上面就出现这个错误，也是因为我对项目进行了更改，我更改了一个完整的网络

下面我就来说一下出现这个问题的原因：

程序中使用了nn.DataParallel，但是PyTorch 中使用 nn.DataParallel 或 nn.parallel.DistributedDataParallel 这样的模型并行技术会导致这个问题，因为它们会将模型包装在另一个模块中，而代码可能在不同的环境下运行

我的修改包括两部分内：

针对代码中， AL_module 是一个模型，直接使用 AL_module 来进行前向传播，无需使用 .module

# 在 train 函数中，将下面这行代码修改：
logits, state = AL_module(net.module.al_input, lab_mask)
# 改为：
logits, state = AL_module(net.al_input, lab_mask)

还有一个原因是：
这种情况通常出现在使用模型并行技术（如 nn.DataParallel 或 nn.parallel.DistributedDataParallel）时，这些技术会在模型外部包装一个额外的模块，从而导致属性访问时出现差异

关注