Spaces:
Running
Running
| """Unified CLI entry point for SAE training.""" | |
| from __future__ import annotations | |
| import sys | |
| from pathlib import Path | |
| from typing import Optional | |
| ROOT = Path(__file__).resolve().parents[1] | |
| if str(ROOT) not in sys.path: | |
| sys.path.insert(0, str(ROOT)) | |
| from log_config import setup_logging | |
| from training.config import parse_args | |
| from training.trainer import Trainer | |
| def main(default_iqa_model: Optional[str] = None) -> None: | |
| setup_logging() | |
| args, backend = parse_args(default_iqa_model=default_iqa_model) | |
| Trainer(backend, args).run() | |
| if __name__ == "__main__": | |
| main() | |