scglue.models.glue.GLUETrainer.val_step

GLUETrainer.val_step(engine, data)[source]

A single validation step

Parameters
Return type

typing.Mapping[str, torch.Tensor]

Returns

loss_dict – Dict containing validation loss values