quaterion.eval.attached_metric 模块

class AttachedMetric(name: str, metric: BaseMetric, logger: bool = True, prog_bar: bool = False, on_step: bool | None = None, on_epoch: bool | None = None, **log_options)

基类: object

将批处理指标附加到 TrainableModel。

包含在训练过程中计算和记录批处理指标所需的参数。

参数:

• name – 要在日志中使用的附加指标的名称。

• metric – 要计算的指标。

• on_step – 在当前步骤记录指标。

• on_epoch – 在 epoch 结束时自动累积和记录。

• prog_bar – 记录到进度条（默认值：False）。

• logger – 记录到日志记录器，例如 Tensorboard，或传递给训练器的任何其他自定义日志记录器（默认值：True）。

• **log_options – 要传递给模型日志方法的附加关键字参数。

• at (其余选项可在) –

• https – //pytorch-lightning.readthedocs.io/en/stable/extensions/logging.html