train404
========

.. py:module:: train404


Attributes
----------

.. autoapisummary::

   train404.description


Classes
-------

.. autoapisummary::

   train404.Objective


Functions
---------

.. autoapisummary::

   train404.setup
   train404.load_dataset_and_sampler
   train404.load_model_states_and_optimizer
   train404.model_and_memory_summary
   train404.main


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

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

.. py:function:: load_dataset_and_sampler(conf, world_size, rank, is_train, seed=42)

.. py:function:: load_model_states_and_optimizer(conf, model, device)

.. py:function:: model_and_memory_summary(conf)

.. py:function:: main(rank, world_size, conf, trial=False)

.. py:class:: Objective(config, metric='val_loss', device='cpu')

   Bases: :py:obj:`echo.src.base_objective.BaseObjective`


   .. py:method:: train(trial, conf)


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