scglue.models.glue.GLUETrainer.val_step

GLUETrainer.val_step(engine, data)

A single validation step

参数:

• engine (ignite.engine.engine.Engine) – Validation engine

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

返回类型:

typing.Mapping[str, torch.Tensor]

返回:

loss_dict – Dict containing validation loss values