Method training_step

locationencoder/imageencoder.py:32–35

(self, batch, batch_idx)

        return self.model(images)

    def training_step(self, batch, batch_idx):
        loss, output = self.common_step(batch, batch_idx)
        self.log("train_loss", loss)
        return loss

    def validation_step(self, batch, batch_idx):
        images, _, labels, _ = batch