Docs
API
pip install pytorch-wrapper
git clone https://github.com/jkoutsikakis/pytorch-wrapper.git cd pytorch-wrapper pip install .