modulation.classification.procedure.single_epoch_fitting

modulation.classification.procedure.single_epoch_fitting(model: Module, optimiser, train_loader_, *, epoch: Optional[int] = None, writer: Optional[Writer] = None, device_: device = device(type='cuda')) → None[source]
Parameters
  • model –

  • optimiser –

  • train_loader –

  • epoch –

  • writer –

  • device –