PyTorch Conv2D Gradient Fix

(Taken from NVIDIA) Replacement for Pytorch's Conv2D and Conv2DTranspose with support for higher-order gradients and disabling unnecessary gradient computations.

Installation

conda install torch-conv-gradfix -c ppeetteerrs

Usage

See the Example tab.