scglue.models.scclue.SCCLUETrainer¶

class scglue.models.scclue.SCCLUETrainer(net, lam_data=None, lam_kl=None, lam_align=None, lam_sup=None, lam_joint_cross=None, lam_real_cross=None, lam_cos=None, normalize_u=None, domain_weight=None, optim=None, lr=None, **kwargs)[source]¶

Bases: scglue.models.base.Trainer

Methods

`compute_losses`	rtype `Mapping`[`str`, `Tensor`]
`fit`	Fit network
`format_data`	rtype `Tuple`[`Mapping`[`str`, `Tensor`], `Mapping`[`str`, `Tensor`], `Mapping`[`str`, `Tensor`], `Mapping`[`str`, `Tensor`], `Mapping`[`str`, `Tensor`], `Mapping`[`str`, `Tensor`], `Tensor`]
`load_state_dict`	Load state from a state dict
`state_dict`	State dict
`train_step`	A single training step
`val_step`	A single validation step

Attributes

logger