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