from dataclasses import dataclass, field
import hydra
from hydra.core.config_store import ConfigStore
from omegaconf import MISSING, OmegaConf
[docs]
@dataclass
class ArgsConfig:
"""
Global arguments configuration.
Defines high-level runtime parameters for experiment execution.
Parameters
----------
dry_run : bool, optional
If True, simulates execution without making changes. Defaults to False.
overwrite : bool, optional
If True, overwrites existing data. Defaults to False.
mode : str, optional
The operation mode (e.g., 'optimize'). Defaults to 'optimize'.
num_trials : int
The number of trials to run.
study_name : str
The name of the study, interpolated from prefix and hydra choices.
max_epochs : int
Maximum number of training epochs.
precision : int, optional
Floating point precision (e.g., 16, 32). Defaults to 32.
accelerator : str, optional
Hardware accelerator to use (e.g., 'gpu', 'cpu'). Defaults to 'gpu'.
"""
dry_run:bool=False
overwrite:bool=False
mode:str='optimize'
num_trials:int=MISSING
study_name:str="${study_prefix}_${hydra:runtime.choices.exp}"
max_epochs:int=MISSING
precision:int=32
accelerator:str="gpu"