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

After Change

        // 2. Compute loss & accuracy:
        train_loss = self.loss(y_logits, y_true)
        accuracy = self.train_acc(y_logits, y_true)

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

Instances