pytorch-lesson-zh

pytorch-lesson-zh copied to clipboard

AlexNet.ipynb FC Layer Param number Wrong

2

hi, When creating Class AlexNet, self.fc goes like this: **************************************** self.fc = nn.Sequential( nn.Linear(1*1*256, 4096), nn.ReLU(inplace=True), nn.Dropout(0.5), nn.Linear(4096, 4096), nn.ReLU(inplace=True), nn.Dropout(0.5), nn.Linear(4096, n_class) # AlexNet上面是1000 ...如果测试的话用MNIST则可以使用10 ) **************************************** but the...

SJLNNU