Correction wrong comments (#1211)
`self.fc1(x)` converts the shape of `x` into "N x 1024", and `self.fc2(x)` converts into "N x num_classes". By adding `print(x.shape)` under each comment line, the console displays as follows (batch_size is 1): ```text torch.Size([1, 2048]) torch.Size([1, 1024]) torch.Size([1, 1024]) torch.Size([1, 1024]) torch.Size([1, 1000]) ```
Showing
Please register or sign in to comment