Huhujingjing
/

custom-gcn

Graph Machine Learning

feature-extraction

Model card Files Files and versions Community

Huhujingjing commited on Jul 1, 2023

Commit

cf18ce2

·

1 Parent(s): 1b1e134

Upload model

Files changed (1) hide show

modeling_gcn.py +1 -1

modeling_gcn.py CHANGED Viewed

@@ -52,7 +52,7 @@ class GCNNet(torch.nn.Module):
         x = scatter(x, batch, dim=-2, reduce='mean')
         x = self.mlp(x)
-        return x.sequeeze(-1)
 class GCNModel(PreTrainedModel):
     config_class = GCNConfig

         x = scatter(x, batch, dim=-2, reduce='mean')
         x = self.mlp(x)
+        return x.squeeze(-1)
 class GCNModel(PreTrainedModel):
     config_class = GCNConfig