cef785da2891f282d420f32df7e5365cf3414a85,torchsample/callbacks.py,TQDM,on_batch_end,#TQDM#Any#Any#,131

Before Change

        self.progbar.update(1)

    def on_batch_end(self, batch, logs=None):
        self.progbar.set_postfix({
            "Loss": "%.04f" % 
            (self.model.history.loss / self.model.history.samples_seen)})


class History(Callback):

After Change

    def on_batch_end(self, batch, logs=None):
        log_data = {key: "%.04f" % value for key, value in self.model.history.batch_metrics.items()}
        for k, v in logs.items():
            if k.endswith("metric"):
                log_data[k.split("_metric")[0]] = "%.02f" % v
        self.progbar.set_postfix(log_data)


class History(Callback):

Instances