model_summary
=============

.. py:module:: model_summary


Attributes
----------

.. autoapisummary::

   model_summary.description


Functions
---------

.. autoapisummary::

   model_summary.setup
   model_summary.main


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

.. py:function:: setup(rank, world_size, mode)

.. py:function:: main(rank, world_size, conf, frames=1, height=640, width=1280)

.. py:data:: description
   :value: 'Train a segmengation model on a hologram data set'