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