--

Hi, Jia. If you check the source code for the layer on GitHub, you will find the corresponding variable names: they are lin_r for W1 matrix and lin_l for W2 matrix. So conv1.lin_l will return a PyTorch instance of the linear layer W1, and then you can see the weights using conv 1.lin_l.weight

--

--

Dmytro Nikolaiev (Dimid)

Machine Learning Enthusiast. Blogging about Data Science / Machine Learning