scglue.models.base.Trainer
- class scglue.models.base.Trainer(net)[source]
Bases:
object
Abstract trainer class
- Parameters:
net (
Module
) – Network module to be trained
Note
Subclasses should populate
required_losses
, and additionally define optimizers here.Methods
Fit network
Get loss values for given data
Load state from a state dict
Report loss values during training
State dict
A single training step
A single validation step
Attributes
logger