48718d7ce74f712da881e9585d66a8e0d54e7004,pl_examples/domain_templates/computer_vision_fine_tuning.py,TransferLearningModel,training_step,#TransferLearningModel#Any#Any#,261

Before Change

        self.train_acc(y_logits, y_true)

        // 3. Outputs:
        tqdm_dict = {"train_loss": train_loss}
        self.log_dict(tqdm_dict, prog_bar=True)
        return {"loss": train_loss}

    def training_epoch_end(self, outputs):

After Change

        train_loss = self.loss(y_logits, y_true)

        // 3. Compute accuracy:
        self.log("train_acc", self.train_acc(y_logits, y_true.int()), prog_bar=True)

        return train_loss

Instances