Offline API Documentation
======================================================================

.. toctree::
   :caption: Contents:

   Trainer <trainer>
   Utils <utils>
   Lightning Callbacks <lightning_callbacks>
   Mine Callbacks <mine_callbacks>

Trainer
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

.. automodule:: minestudio.offline.trainer
   :members:

Utils
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

.. automodule:: minestudio.offline.utils
   :members:

Lightning Callbacks
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

EMA Callback
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. automodule:: minestudio.offline.lightning_callbacks.ema
   :members:

Smart Checkpoint Callback
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. automodule:: minestudio.offline.lightning_callbacks.smart_checkpoint
   :members:

Speed Monitor Callback
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. automodule:: minestudio.offline.lightning_callbacks.speed_monitor
   :members:

Mine Callbacks
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

Behavior Clone Callback
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. automodule:: minestudio.offline.mine_callbacks.behavior_clone
   :members:

Callback
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. automodule:: minestudio.offline.mine_callbacks.callback
   :members:

KL Divergence Callback
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. automodule:: minestudio.offline.mine_callbacks.kl_divergence
   :members: