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 –