def METHOD_NAME(self, outputs):
    for name, mf in self.train_metrics.items():
        self.log(f'train_{name}', mf.compute(), prog_bar=False)
    for name, mf in self.train_metrics.items():
        mf.reset()