spyrit.core.train.tb_writer_add_scalar

spyrit.core.train.tb_writer_add_scalar(writer, name_metric, val_metric, step)[source]

Tensorboard writer: Add a scalar (loss)