以下模型返回错误: 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 ×1