GitHub repository: mit-han-lab/patch_conv
Primary language: Python
Description: Patch convolution to avoid large GPU memory usage of Conv2D