wehub-resource-sync
17 lines
583 B
Python
17 lines
583 B
Python
# Copyright Lightning AI. Licensed under the Apache License 2.0, see LICENSE file.
|
|
"""Type aliases used across LitGPT modules."""
|
|
|
|
from typing import Literal
|
|
|
|
# Logger-related types
|
|
LoggerChoice = Literal["csv", "tensorboard", "wandb", "mlflow", "litlogger"]
|
|
"""Valid logger choices for experiment tracking.
|
|
|
|
Available options:
|
|
- "csv": Local CSV file logging (default for most scripts)
|
|
- "tensorboard": TensorBoard visualization (default for pretrain)
|
|
- "wandb": Weights & Biases cloud tracking
|
|
- "mlflow": MLflow experiment tracking
|
|
- "litlogger": Lightning.ai native tracking
|
|
"""
|