scglue.models.scclue.SCCLUETrainer.val_step
- SCCLUETrainer.val_step(engine, data)[source]
A single validation step
- Parameters
engine (
ignite.engine.engine.Engine
) – Validation enginedata (
typing.List
[torch.Tensor
]) – Data of the validation step
- Return type
- Returns
loss_dict – Dict containing validation loss values