Thanks for the code!
In your first example you are passing the batch_size=bs directly to sumamry(), which will then print it out.
If you skip it and leave it as -1 (default), you’ll get:
----------------------------------------------------------------
        Layer (type)               Output Shape         Param #
================================================================
            Linear-1                    [-1, 7]          86,023
            Linear-2                    [-1, 5]              40
            Linear-3                    [-1, 2]              12
================================================================
Total params: 86,075
Trainable params: 86,075
Non-trainable params: 0
----------------------------------------------------------------
Input size (MB): 0.05
Forward/backward pass size (MB): 0.00
Params size (MB): 0.33
Estimated Total Size (MB): 0.38
----------------------------------------------------------------