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

         @hydra.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