0e8ec93b28ecd86898b3668d48d346f510e95167,pl_examples/domain_templates/computer_vision_fine_tuning.py,TransferLearningModel,training_step,#TransferLearningModel#Any#Any#,260

Before Change

        // 3. Outputs:
        tqdm_dict = {"train_loss": train_loss}
        output = OrderedDict({"loss": train_loss,
                              "num_correct": num_correct,
                              "log": tqdm_dict,
                              "progress_bar": tqdm_dict})

        return output

    def training_epoch_end(self, outputs):
        Compute and log training loss and accuracy at the epoch level.

After Change

        // 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):
        Compute and log training loss and accuracy at the epoch level.

Instances