from pydantic.dataclasses import dataclass
from ..base.evaluator_config import EvaluatorConfig
[docs]
@dataclass
class VisualizationConfig(EvaluatorConfig):
"""Config class for the visualization module.
Args:
batch_size (int) : The batch size to use in the evaluation. Default to 20
wandb_path (str) : The user can provide the path of the wandb run with a
format 'entity/projet_name/run_id' where the metrics should be logged.
See :doc:`info_wandb` for more information.
If None is provided, the metrics are not logged on wandb.
Default to None.
n_samples(int) : The number of samples to generate per modality and per data_point for
conditional generation. Default to 5.
n_data_cond (int) : The number of datapoints to use for conditional generation. Default to 5
"""
batch_size: int = 20
n_samples: int = 5
n_data_cond: int = 5