SRGAN出现AttributeError: 'Generator' object has no attribute 'module'

这个错误通常出现在使用 PyTorch 训练和保存模型时。`AttributeError: 'Generator' object has no attribute 'module'` 错误表示在尝试访问生成器对象的 'module' 属性时发生了错误。

这个问题通常是由于在加载模型时，模型被包装在 `nn.DataParallel` 或 `nn.parallel.DistributedDataParallel` 中，而不是直接的模型对象。这会导致生成器对象没有 'module' 属性。

解决这个问题的方法是在访问生成器对象之前，先对其进行处理以获取内部的模型。以下是一个示例代码：

import torch

# 加载生成器模型
generator = torch.load('generator_model.pth')

# 检查是否使用了 DataParallel 或 DistributedDataParallel
if isinstance(generator, torch.nn.DataParallel) or isinstance(generator, torch.nn.parallel.DistributedDataParallel):
    generator = generator.module  # 获取内部模型

# 使用生成器进行预测或其他操作
output = generator(input)

在示例代码中，我们首先加载了生成器模型。然后，我们检查生成器是否被包装在 `DataParallel` 或 `DistributedDataParallel` 中。如果是，我们通过访问 `.module` 属性来获取内部的模型。最后，我们可以使用生成器进行预测或其他操作。

请根据你的具体情况修改代码，并确保模型的加载和访问的一致性。