scglue.models.base.Trainer.train_step
- abstract Trainer.train_step(engine, data)[源代码]
A single training step
- 参数:
engine (
ignite.engine.engine.Engine
) – Training enginedata (
typing.List
[torch.Tensor
]) – Data of the training step
- 返回类型:
- 返回:
loss_dict – Dict containing training loss values