basicv8vc / backpack

BACKpropagation PACKage - A backpack for PyTorch that extends the backward pass of feedforward networks to compute quantities beyond the gradient

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

BACKpropagation PACKage - a backpack for PyTorch

A backpack for PyTorch that extends the backward pass of feedforward networks to compute quantities beyond the gradient.

Check out the examples on how to use the code.

Installation

pip install backpack-for-pytorch

Alternative: Clone the repository and run setup.py

git clone https://github.com/f-dangel/backpack.git ~/backpack
cd ~/backpack
python setup.py install

About

BACKpropagation PACKage - A backpack for PyTorch that extends the backward pass of feedforward networks to compute quantities beyond the gradient

License:MIT License


Languages

Language:Python 87.2%Language:HTML 7.3%Language:CSS 3.5%Language:Ruby 0.6%Language:Shell 0.5%Language:JavaScript 0.4%Language:Batchfile 0.3%Language:Makefile 0.3%