Hey It seems to run now after changing the last line of linear code into self.fc1 = nn.Linear(4096, 2) and adding this
output = output.view(output.size(0), -1) in forward. Could you please explain what exactly does the output.view line of code does?
Thanks so much for this help.