(self, x, edge_index, edge_weigt=None, batch=None)
| 111 | |
| 112 | |
| 113 | def forward(self, x, edge_index, edge_weigt=None, batch=None): |
| 114 | |
| 115 | h = self.encoder(x, edge_index, edge_weigt, batch) |
| 116 | h = self.proj_head(h) |
| 117 | |
| 118 | return h |
| 119 | |
| 120 | |
| 121 | def get_loss(self, x, x_aug, batch_size): |
nothing calls this directly
no outgoing calls
no test coverage detected