ac7b6cf0992219b5118738e05672306f8acc215f,catalyst/dl/state.py,RunnerState,on_batch_end_pre,#RunnerState#,160

Before Change

        elapsed_time = time.time() - self._datatime

        self.batch_metrics["base/batch_time"] = elapsed_time
        self.batch_metrics["base/sample_per_second"] = \
            self.batch_size / elapsed_time

    def on_batch_end_post(self):
        lm = self.loader_name
        for key, value in self.batch_metrics.items():

After Change

    def on_batch_end_pre(self):
        elapsed_time = time.time() - self._datatime

        self.metrics.add_batch_value("base/batch_time", elapsed_time)
        self.metrics.add_batch_value(
            "base/sample_per_second", self.batch_size / elapsed_time
        )

Instances