Contents:
IndSCGLUETrainer.
train_step
A single training step
engine (Engine) – Training engine
Engine
data (List[Tensor]) – Data of the training step
List
Tensor
Mapping[str, Tensor]
Mapping
str
loss_dict – Dict containing training loss values