小编Roo*_*ter的帖子

自动编码器 MaxUnpool2d 缺少“索引”参数

以下模型返回错误: TypeError:forward() Missing 1 requiredpositional argument: 'indices'

我已经用尽了许多在线示例,它们看起来都与我的代码相似。我的 maxpool 层返回 unpool 层的输入和索引。关于出了什么问题有什么想法吗?

class autoencoder(nn.Module):
def __init__(self):
    super(autoencoder, self).__init__()
    self.encoder = nn.Sequential(
        ...
        nn.MaxPool2d(2, stride=1, return_indices=True)
    )
    self.decoder = nn.Sequential(
        nn.MaxUnpool2d(2, stride=1),
        ...
    )

def forward(self, x):
    x = self.encoder(x)
    x = self.decoder(x)
    return x
Run Code Online (Sandbox Code Playgroud)

pytorch

4
推荐指数
1
解决办法
4944
查看次数

标签 统计

pytorch ×1