Lion
====

.. currentmodule:: lib.model.optimizers.lion

.. autoclass:: Lion
   :show-inheritance:
   :noindex:

   .. rubric:: Attributes Summary

   .. autosummary::

      ~Lion.OptimizerPostHook
      ~Lion.OptimizerPreHook

   .. rubric:: Methods Summary

   .. autosummary::

      ~Lion.add_param_group
      ~Lion.load_state_dict
      ~Lion.profile_hook_step
      ~Lion.register_load_state_dict_post_hook
      ~Lion.register_load_state_dict_pre_hook
      ~Lion.register_state_dict_post_hook
      ~Lion.register_state_dict_pre_hook
      ~Lion.register_step_post_hook
      ~Lion.register_step_pre_hook
      ~Lion.state_dict
      ~Lion.step
      ~Lion.zero_grad

   .. rubric:: Attributes Documentation

   .. autoattribute:: OptimizerPostHook
   .. autoattribute:: OptimizerPreHook

   .. rubric:: Methods Documentation

   .. automethod:: add_param_group
   .. automethod:: load_state_dict
   .. automethod:: profile_hook_step
   .. automethod:: register_load_state_dict_post_hook
   .. automethod:: register_load_state_dict_pre_hook
   .. automethod:: register_state_dict_post_hook
   .. automethod:: register_state_dict_pre_hook
   .. automethod:: register_step_post_hook
   .. automethod:: register_step_pre_hook
   .. automethod:: state_dict
   .. automethod:: step
   .. automethod:: zero_grad