CondAEModel.test

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

Evaluate the model on the test data.

Parameters:
  • datamodule (DataModule) – A compatible data module object that provides test data.

  • accelerator (str, optional, default=”auto”) – Which accelerator should be used (e.g. cpu, gpu, mps, etc.).

Returns:

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