def on_loader_start(self, runner): """ Calls runner methods when the dataloader begins and adds metrics for loss and macro_dice """ super().on_loader_start(runner) self.meters = { key: metrics.AdditiveValueMetric(compute_on_call=False) for key in ["loss", "macro_dice"] }
def on_loader_start(self, runner): super().on_loader_start(runner) self.meters = { key: metrics.AdditiveValueMetric(compute_on_call=False) for key in ["loss", "accuracy01", "accuracy03"] }
def on_loader_start(self, runner: dl.IRunner): super().on_loader_start(runner) self.meters = { key: metrics.AdditiveValueMetric(compute_on_call=False) for key in ["critic_loss", "actor_loss"] }