In-Place Activated BatchNorm for Memory-Optimized Training of DNNs
conda install conda-forge::inplace-abn