I created a NN model in PyTorch.
This is my model summary:
Training time: 2939.56
Layer (type) Output Shape Param #
================================================================
Linear-1 [-1, 1, 10, 10] 30
Tanh-2 [-1, 1, 10, 10] 0
Linear-3 [-1, 1, 10, 10] 110
Tanh-4 [-1, 1, 10, 10] 0
Linear-5 [-1, 1, 10, 10] 110
Tanh-6 [-1, 1, 10, 10] 0
Linear-7 [-1, 1, 10, 10] 110
Tanh-8 [-1, 1, 10, 10] 0
Linear-9 [-1, 1, 10, 1] 11
Total params: 371
Trainable params: 371
Non-trainable params: 0
Input size (MB): 0.00
Forward/backward pass size (MB): 0.01
Params size (MB): 0.00
Estimated Total Size (MB): 0.01
But what does this mean? What does it tell me? How can I interpret it?
Thanks guys 