CondAEModel.validate

CondAEModel.validate(datamodule: DataModule, accelerator: str = 'auto', **kwargs) Dict[str, float][source]

Evaluate the model on the validation data.

Parameters:
  • datamodule (DataModule) – The data module object that provides validation data.

  • accelerator (str, optional, default=”auto”) – Supports passing different accelerator types (“cpu”, “gpu”, “tpu”, “ipu”, “hpu”, “mps”, “auto”).

Returns:

Dict[str, float] – A dictionary containing the validation loss and metrics.