index.rst

Welcome to torch-influence's API Reference!

Table of Contents

• Base Modules
• Influence Modules

.. currentmodule:: torch_influence

Base Modules

.. autoclass:: BaseInfluenceModule
   :members:

.. autoclass:: BaseObjective
   :members:

Influence Modules

torch-influence provides three subclasses of :class:`BaseInfluenceModule` out-of-the-box. Each subclass differs only in how the abstract function :meth:`BaseInfluenceModule.inverse_hvp()` is implemented. We refer readers to the original influence function paper (Koh & Liang, 2017) for further details.

.. autoclass:: AutogradInfluenceModule
   :show-inheritance:

.. autoclass:: CGInfluenceModule
   :show-inheritance:

.. autoclass:: LiSSAInfluenceModule
   :show-inheritance: