custom_scalars

class AddMetricAndLossCustomScalars[source]

Add custom scalars to the tensorboard logger, for comparison of train and val metrics and losses.

on_fit_start(trainer: Trainer, pl_module: LightningModule) → None[source]

Adds the custom scalars.