I use notron to visualize my model, but I dont know how to compute the W(weights) and B(bias) ,I use formula, but it dont work. as you can see, how to compute the second Conv structure from the first Conv stucture?

The `in_channels`

of the second convolution would be defined by the `out_channels`

of the previous one, which defines the size of `dim1`

(32 in your example). Besides that you can pick other parameters as you wish.

1 Like