a0ad3b8ab0666bff2bdf2028a0a6aa8e559d5bc1,steps/pytorch/callbacks.py,ExperimentTiming,on_epoch_begin,#ExperimentTiming#,341

Before Change

    def on_epoch_begin(self, *args, **kwargs):
        if self.epoch_id > 0:
            epoch_time = datetime.now() - self.epoch_start
            logger.info("epoch {0} time {1}".format(self.epoch_id - 1, str(epoch_time)[:-7]))
        self.epoch_start = datetime.now()
        self.current_sum = timedelta()
        self.current_mean = timedelta()
        logger.info("epoch {0} ...".format(self.epoch_id))

After Change

        if self.epoch_id > 0:
            epoch_time = datetime.now() - self.epoch_start
            if self.epoch_every:
                if (self.epoch_id % self.epoch_every) == 0:
                    logger.info("epoch {0} time {1}".format(self.epoch_id - 1, str(epoch_time)[:-7]))
        self.epoch_start = datetime.now()
        self.current_sum = timedelta()
        self.current_mean = timedelta()
        logger.info("epoch {0} ...".format(self.epoch_id))

Instances