Artificial Intelligence, Machine Learning, Deep Learning, Image Classification, Heartbeat.
output=self.pretrain_model(input) return output def training_step(self, batch, batch_idx): outputs, targets, loss, preds = self._common_step(batch, batch_idx) train_accuracy = self.accuracy(preds, targets) self.history['train_loss'].append(loss.item())