neuro_morpho.cli
================

.. py:module:: neuro_morpho.cli

.. autoapi-nested-parse::

   Command line interface for training and running models.



Functions
---------

.. autoapisummary::

   neuro_morpho.cli.register_torch_transforms
   neuro_morpho.cli.main


Module Contents
---------------

.. py:function:: register_torch_transforms() -> None

   Register torch transforms to gin.

   This allows the user to configure torchvision transforms from the gin config file.


.. py:function:: main(config: str = 'unet.config.gin') -> None

   Run the training and inference pipeline.

   :param config: The path to the gin configuration file.
   :type config: str