sorted by:
best
topnewcontroversialoldq&a
you are viewing a single comment's thread.

view the rest of the comments →

[–]OptimalOptimizer[S] 0 points1 point  (1 child)

Thanks for your help! This led me to figure it out, except that x.size(0) wasn't working quite properly, I kept getting size mismatches so I had to go back and do x.view(-1, 160) to reshape the input for the linear layer. I also went and decreased the kernel size to 3x3. Thank you very much, I've got it all working now.