I am not sure, you should try and print the size of the tensor that you give as input of your forward pass and ensure that its size is (batch x 3 x 299 x 299) (even if batch is 1, it has to be there).