Hello, I have modified the output of the Avg Pool layer before the input layer and now the network works, so the first solution is great. Thanks a lot!!!
I also tried the second soulution of changing the input of 73728 - Linear(in_features=73728, out_features=1000, bias=True) , but it was too large and my computer cannot works.