a1cdc5d5b8bc04f2aa8d253c52f149c28ddd8f21,base/base_trainer.py,BaseTrainer,train,#BaseTrainer#,58

Before Change

            log = {"epoch": epoch}
            for key, value in result.items():
                if key == "metrics":
                    for i, metric in enumerate(self.metrics):
                        log[metric.__name__] = result["metrics"][i]
                elif key == "val_metrics":
                    for i, metric in enumerate(self.metrics):
                        log["val_" + metric.__name__] = result["val_metrics"][i]
                else:

After Change

                if key == "metrics":
                    log.update({mtr.__name__ : value[i] for i, mtr in enumerate(self.metrics)})
                elif key == "val_metrics":
                    log.update({"val_" + mtr.__name__ : value[i] for i, mtr in enumerate(self.metrics)})
                else:
                    log[key] = value
            if self.train_logger is not None:
                self.train_logger.add_entry(log)

Instances