I’m getting confused. After your first comment, I came across above implementation, which I now reckon is nothing but an.Conv2d and mn.Batchnorm2d.
For the first point you mentioned, should I inherit the Conv2d and B_Norm in super(Net,self).init(Conv2d,B_Norm) like this?
If so then it will show error.
But yes, I want to only use the functional API. Which I tried, I created parameters (self.conv_1_weight,…), but I really don’t get how can I create buffers and how can I store running stats for F.batch_norm as well.
Can you please show, how it can be done?
Many thanks.