I modify some layers and I want to create a class for them. For example,
class mod:
class linear(nn.Module):
def __init__(self, d_int, d_out):
pass
def forward(self, x):
pass
When I try to initialize this layer by
layer = mod.linear(2, 3)
I get an error saying
super(type, obj): obj must be an instance or subtype of type
Is there anyway to solve this problem?