CondAEModel.test

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

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.