(self, lonlats)
| 98 | return self.loss_fn(self, lonlats, label) |
| 99 | |
| 100 | def forward(self, lonlats): |
| 101 | embedding = self.positional_encoder(lonlats) |
| 102 | return self.neural_network(embedding) |
| 103 | |
| 104 | def training_step(self, batch, batch_idx): |
| 105 | loss = self.common_step(batch, batch_idx) |
no outgoing calls