Update models.py

Author: CurryTang

models.py

@@ -164,7 +164,10 @@ def __init__(self, num_layers, input_dim, hidden_dimension, num_classes, dropout
             for _ in range(num_layers - 2):
                 self.convs.append(GCNConv(hidden_dimension, hidden_dimension, cached=False,
                              normalize=True))
-                self.norms.append(torch.nn.BatchNorm1d(hidden_dimension))
+                if norm:
+                    self.norms.append(torch.nn.BatchNorm1d(hidden_dimension))
+                else:
+                    self.norms.append(torch.nn.Identity())
 
             self.convs.append(GCNConv(hidden_dimension, num_classes, cached=False, normalize=True))
 
@@ -599,4 +602,4 @@ def forward(self, graph, feat):
         h = self.bias_last(h)
 
         return h
-               
+