[Model] Update GraphSAGE (#384)

* [Feature] Small Changes * [Feature] Update Graphsage
THUDM · Sep 2, 2022 · c3d23bb · c3d23bb
1 parent 4c8737c
commit c3d23bb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cogdl/models/nn/graphsage.py b/cogdl/models/nn/graphsage.py
@@ -85,6 +85,7 @@ def mini_forward(self, graph):
             if i != self.num_layers - 1:
                 x = F.relu(x)
                 x = F.dropout(x, p=self.dropout, training=self.training)
+        self.adjlist.clear()
         return x
 
     def forward(self, *args):