RuntimeError: shape '[-1, 576]' is invalid for input of size 12800

4

Thanks you for the solution! However, I want to know how to get the output shape of each layer. How do you find the output shape of each layer?