scglue.models.scglue.SCGLUETrainer.train_step

SCGLUETrainer.train_step(engine, data)[source]

A single training step

Parameters

  • engine (Engine) – Training engine

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

Return type

Mapping[str, Tensor]

Returns

loss_dict – Dict containing training loss values