plugins package

Subpackages

• plugins.natgrad package
  • Submodules
  • plugins.natgrad.ekfac module
    • EKFAC
    • grad_wrt_kernel()
  • plugins.natgrad.kfac module
    • KFAC
  • Module contents

Submodules

plugins.ekfac_plugin module

class plugins.ekfac_plugin.EKFAC_Plugin(network, eps=0.1)

Bases: BaseSGDPlugin

before_update(strategy)

Called before optimizer.update() by the BaseTemplate.

class plugins.ekfac_plugin.KFAC_Plugin(network, eps=0.1)

Bases: BaseSGDPlugin

before_update(strategy)

Called before optimizer.update() by the BaseTemplate.