I am trying to get a good summary of my deep learning model like

So, I am not sure what values should I put in the input_size parameter. Can anyone please help me find the issue or find my actual input shape for the model for getting a summary?

CodePudding user response：

Linear expects the number of channels on the last axis of the input to be in_features.

model_stats = summary(my_model, input_size=(1, 8, 34))