.. _training_step:

.. module:: dipm.training.training_step

Training step
=============

.. autoclass:: TrainingState

.. autofunction:: make_train_step