sccross.models.sccross.SCCROSSTrainer.val_step

SCCROSSTrainer.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