Thanks for your response!
I think you’re right – view is better than squeeze in this case as my batch_size>1. So thank you for pointing this out.
When I retrained the model with view, interestingly enough, I got the same accuracy as what I observed with squeeze. I’m not sure why this was the case (any ideas?), but nonetheless thank you for helping me make my code better reflect my intentions.