modulation.classification.procedure.single_epoch_evaluation

modulation.classification.procedure.single_epoch_evaluation(model: Module, evaluation_loader: DataLoader, subset: SplitEnum, *, epoch: Optional[int] = None, writer: Optional[Writer] = None, device: device = device(type='cuda')) float[source]
Parameters
  • model

  • evaluation_loader

  • subset

  • epoch

  • writer

  • device

Returns

Return type