scglue.models.base.Trainer.report_metrics

Trainer.report_metrics(train_state, val_state)[source]

Report loss values during training

Parameters

  • train_state (State) – Training engine state

  • val_state (Optional[State]) – Validation engine state

Return type

None