Method common_step

locationencoder/imageencoder.py:24–27

(self, batch, batch_idx)

        self.save_hyperparameters()

    def common_step(self, batch, batch_idx):
        images, _, labels, _ = batch
        output = self.model(images)
        return self.criterion(output, labels), output

    def forward(self, images):
        return self.model(images)