Source code for power_cogs.config.tune.tune_config
from dataclasses import field
from typing import Any, Dict, Optional
from omegaconf import II
from pydantic.dataclasses import dataclass
[docs]@dataclass
class TuneConfig:
metric: str = "loss"
mode: str = "min"
num_samples: int = 1
name: Optional[str] = II("trainer.name")
checkpoint_freq: int = 100
checkpoint_at_end: bool = True
additional_config: Dict[str, Any] = field(default_factory=lambda: {})