update small benchmark

small_benchmark/main.py

@@ -87,7 +87,7 @@ def main(conf):
         num_nodes=data.num_nodes,
         in_channels=in_channels,
         out_channels=out_channels,
-        device=device,  # Put histories on GPU.
+        device=device,  # ... and put histories on GPU as well.
         **params.architecture,
     ).to(device)
 

torch_geometric_autoscale/models/gat.py

@@ -43,8 +43,6 @@ class GAT(ScalableGNN):
         super().reset_parameters()
         for conv in self.convs:
             conv.reset_parameters()
-        for lin in self.lins:
-            lin.reset_parameters()
 
     def forward(self, x: Tensor, adj_t: SparseTensor, *args) -> Tensor:
         for conv, history in zip(self.convs[:-1], self.histories):