(self, batch, batch_idx)
| 111 | return {"val_loss":loss} |
| 112 | |
| 113 | def predict_step(self, batch, batch_idx): |
| 114 | lonlats, label = batch |
| 115 | prediction_logits = self.forward(lonlats) |
| 116 | return prediction_logits, lonlats, label |
| 117 | |
| 118 | def test_step(self, batch, batch_idx): |
| 119 | lonlats, label = batch |