Spaces:
Sleeping
Sleeping
| Setup Config | |
| ============ | |
| To set up your configuration, you will need to generate a configuration class based on :class:`~yolo.config.config.Config`, which can be achieved using `hydra <https://hydra.cc/>`_. | |
| The configuration will include all the necessary settings for your ``task``, including general configuration, ``dataset`` information, and task-specific information (``train``, ``inference``, ``validation``). | |
| Next, create the progress logger to handle the output and progress bar. This class is based on `rich <https://github.com/Textualize/rich>`_'s progress bar and customizes the logger (print function) using `loguru <https://loguru.readthedocs.io/>`_. | |
| .. tabs:: | |
| .. tab:: decorator | |
| .. code-block:: python | |
| import hydra | |
| from yolo import ProgressLogger | |
| from yolo.config.config import Config | |
| .main(config_path="config", config_name="config", version_base=None) | |
| def main(cfg: Config): | |
| progress = ProgressLogger(cfg, exp_name=cfg.name) | |
| pass | |
| .. tab:: initialize & compose | |
| .. code-block:: python | |
| from hydra import compose, initialize | |
| from yolo import ProgressLogger | |
| from yolo.config.config import Config | |
| with initialize(config_path="config", version_base=None): | |
| cfg = compose(config_name="config", overrides=["task=train", "model=v9-c"]) | |
| progress = ProgressLogger(cfg, exp_name=cfg.name) | |
| TODO: add a config over view | |