Closed ColdAsYou165 closed 3 years ago
self.fc1 = nn.Linear(8 * 8 * out_channel2, 1024) conv2的输出应为7*7*out_channel2,所以应将上述代码的8改为7 (菜鸡一枚,班门弄斧了)
self.fc1 = nn.Linear(8 * 8 * out_channel2, 1024)
Thank you for opened this issue. I've updated this file.
self.fc1 = nn.Linear(8 * 8 * out_channel2, 1024)
conv2的输出应为7*7*out_channel2,所以应将上述代码的8改为7 (菜鸡一枚,班门弄斧了)