Hi, thank you for your answer but isn’t the line
hidden = [nn.Linear(self.output_conv, hidden_dim) if i==0 else nn.Linear(hidden_dim, hidden_dim) for i in range(hidden_num)]

generating a line of linear layers with the first layer with input size self.output_conv? Or am I missing something?
Thank you once again.