TAAFT
Free mode
100% free
Freemium
Free Trial
Deals

K-FAC: Kronecker-Factored Approximate Curvature

tensorflow / kfac

An implementation of KFAC for TensorFlow

199 41 Language: Python License: Apache-2.0 Updated: 5mo ago

README

K-FAC: Kronecker-Factored Approximate Curvature

Travis

K-FAC in TensorFlow is an implementation of K-FAC, an
approximate second-order optimization method, in TensorFlow.

Installation

kfac is compatible with Python 2 and 3 and can be installed directly via
pip,

# Assumes tensorflow or tensorflow-gpu installed
$ pip install kfac

# Installs with tensorflow-gpu requirement
$ pip install 'kfac[tensorflow_gpu]'

# Installs with tensorflow (cpu) requirement
$ pip install 'kfac[tensorflow]'

KFAC DOCS

Please check KFAC docs for a detailed description with examples
of how to use KFAC. Check the Keras KFAC docs for information on
using KFAC with Keras.

0 AIs selected
Clear selection
#
Name
Task