scglue.models.glue.GLUETrainer¶
-
class
scglue.models.glue.
GLUETrainer
(net, lam_data=None, lam_kl=None, lam_graph=None, lam_align=None, domain_weight=None, optim=None, lr=None, **kwargs)[source]¶ Bases:
scglue.models.base.Trainer
Trainer for
GLUE
- Parameters
Methods
Compute loss functions
Fit network
Format data tensors
Get loss values for given data
Load state from a state dict
State dict
A single training step
A single validation step
Attributes
logger