model not showing the parameters

[ghost]

class CircleModel(nn.Module):
    def __init__(self):
       super().__init__()
       
       self.layer_1 = nn.Linear(in_features=2, out_features=10)
       self.layer_2 = nn.Linear(in_features=10, out_features=10)
       self.layer_3 = nn.Linear(in_features=10, out_features=1)

model = CircleModel()

• While checking the model . The model is not showing the parameters its says empty
• Also while optimizing it says parameters cannot be empty

How to fix It ?

[tonyhaya]

You must write the forward() and return your layer inside of it, like this:

class CicleModelV2(nn.Module):
  def __init__(self):
    super().__init__()
    
    self.layer1=nn.Sequential(
      nn.Linear(2,10),
      nn.Linear(10,10),
      nn.Linear(10,1))

   # Required forward() method
  def forward(self, x):
    return self.layer1(x)

[ghost]

Thank you bro