Cannot see a Parameter in the model.

[businessabstraction]

When I create a Parameter in __init__()

class maxSpeedGNN(torch.nn.Module):
    """ Network to guess the hidden values and the maximums """ 
    def __init__(self, data, report=False):
        super(maxSpeedGNN, self).__init__()
        self.i2o = In2Out()        
        input_num = data['In'].x.size()[0]
        self.input_embedding = torch.nn.Parameter(torch.zeros(input_num, INPUT_DIM))
        torch.nn.init.uniform_(self.input_embedding)          
        return
        
    def forward(self, data, report=False):        
        outputs = self.i2o(data['In', 'Out'].edge_index, data['In', 'Out'].sz, x=self.input_embedding, report=report)     
        return outputs

It doesn't show:

model
---------
maxSpeedGNN(
  (i2o): In2Out()
)

[akihironitta]

Yes, that's expected. You can still check parameters by running:

for i in model.parameters():
    print(i)