scglue.models.glue.GLUETrainer
- class scglue.models.glue.GLUETrainer(net, lam_data=None, lam_kl=None, lam_graph=None, lam_align=None, modality_weight=None, optim=None, lr=None, **kwargs)[源代码]
基类:
Trainer
Trainer for
GLUE
- 参数:
net (
scglue.models.glue.GLUE
) –GLUE
network to be trainedlam_data (
typing.Optional
[float
]) – Data weightlam_graph (
typing.Optional
[float
]) – Graph weightlam_align (
typing.Optional
[float
]) – Adversarial alignment weightmodality_weight (
typing.Optional
[typing.Mapping
[str
,float
]]) – Relative modality weight (indexed by modality name)optim (
typing.Optional
[str
]) – Optimizerlr (
typing.Optional
[float
]) – Learning rate**kwargs – Additional keyword arguments are passed to the optimizer constructor
方法
Compute loss functions
训练模型
Format data tensors :rtype:
typing.Tuple
[typing.Mapping
[str
,torch.Tensor
],typing.Mapping
[str
,torch.Tensor
],torch.Tensor
,torch.Tensor
,torch.Tensor
]Get loss values for given data
Load state from a state dict
State dict
A single training step
A single validation step
属性
logger