self.layers.append(NodeApplyModule(n_hidden, n_classes))
def forward(self, features):
self.g.set_n_repr(features)
for layer in self.layers:
// apply dropout
if self.dropout:
val = F.dropout(self.g.get_n_repr(), p=self.dropout)
After Change
self.layers.append(NodeApplyModule(n_hidden, n_classes))
def forward(self, features):
self.g.set_n_repr({"h" : features})
for layer in self.layers:
// apply dropout
if self.dropout: