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