(self, graph)
| 52 | ) |
| 53 | |
| 54 | def mini_forward(self, graph): |
| 55 | x = graph.x |
| 56 | for i in range(self.num_layers): |
| 57 | edge_index_sp = self.sampling(graph.edge_index, self.sample_size[i]).to(x.device) |
| 58 | with graph.local_graph(): |
| 59 | graph.edge_index = edge_index_sp |
| 60 | x = self.convs[i](graph, x) |
| 61 | if i != self.num_layers - 1: |
| 62 | x = F.relu(x) |
| 63 | x = F.dropout(x, p=self.dropout, training=self.training) |
| 64 | return x |
| 65 | |
| 66 | def forward(self, *args): |
| 67 | if isinstance(args[0], Graph): |
no test coverage detected