Number of activations for Linear and Conv2d layer: comparison

[blackbirdbarber (bbb)]

Tested and it works great. Thanks.