scglue.models.scclue.SCCLUETrainer.train_step
- SCCLUETrainer.train_step(engine, data)[source]
A single training step
- Parameters
engine (
ignite.engine.engine.Engine
) – Training enginedata (
typing.List
[torch.Tensor
]) – Data of the training step
- Return type
- Returns
loss_dict – Dict containing training loss values