scglue.models.glue.GLUETrainer.train_step

GLUETrainer.train_step(engine, data)

A single training step

Parameters:

• engine (ignite.engine.engine.Engine) – Training engine

• data (typing.List[torch.Tensor]) – Data of the training step

Return type:

typing.Mapping[str, torch.Tensor]

Returns:

loss_dict – Dict containing training loss values