scglue.models.glue.GLUETrainer.train_step
- GLUETrainer.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