Solved by Usama_Hasan in post #8 
              
              
                Hy @Kunal_Dapse, I would highly recommend you read some tutorials first, you’re totaly misunderstanding me here. The way we construct models in pytorch is by inheriting them through nn.Module class. 
Something like this 
import torch.nn as nn
class Net(nn.Module):
       nn.Linear(input_channels,out…