scglue.models.scclue.SCCLUETrainer.val_step

SCCLUETrainer.val_step(engine, data)[source]

A single validation step

Parameters

  • engine (Engine) – Validation engine

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

Return type

Mapping[str, Tensor]

Returns

loss_dict – Dict containing validation loss values