Hi @Pomo,
Thanks for responding, I did not know about the model.summary(), thanks!
I am Getting this error
TypeError: forward() missing 1 required positional argument: ‘input’
but only happens with the 201, with the other ones 161, etc i dont get this error.