utils.rst

Utilities and helper functions

This subpackage contains common helper functions to a variety of problems (e.g., PyTorch checkpointing, special layers, computing diagonal Fisher matrices, ...).

.. automodule:: hypnettorch.utils.batchnorm_layer
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: hypnettorch.utils.cli_args
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: hypnettorch.utils.context_mod_layer
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: hypnettorch.utils.ewc_regularizer
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: hypnettorch.utils.gan_helpers
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: hypnettorch.utils.hmc
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: hypnettorch.utils.hnet_regularizer
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: hypnettorch.utils.init_utils
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: hypnettorch.utils.local_conv2d_layer
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: hypnettorch.utils.logger_config
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: hypnettorch.utils.misc
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: hypnettorch.utils.optim_step
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: hypnettorch.utils.self_attention_layer
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: hypnettorch.utils.si_regularizer
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: hypnettorch.utils.sim_utils
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: hypnettorch.utils.torch_ckpts
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: hypnettorch.utils.torch_utils
    :members:
    :undoc-members:
    :show-inheritance: