File size: 601 Bytes
feba2ad | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | """
Monitoring Config
Specifies the monitoring process, e.g. how to log metrics and keep track of training progress.
"""
from dataclasses import dataclass, field
@dataclass
class LoggingConfig:
log_level: str = "INFO"
log_every_n_steps: int = 100
@dataclass
class WandbConfig:
# configure logging to Weights and Biases
project: str = ""
entity: str = ""
@dataclass
class MonitoringConfig:
logging: LoggingConfig = field(default_factory=LoggingConfig)
# Weights and Biases
save_to_wandb: bool = False
wandb: WandbConfig = field(default_factory=WandbConfig)
|