IndexedConv v1.1
Mikaël Jacquemont, Thomas Vuillaume · Figshare · 2019
# IndexedConv v1.1 release Convolution memory consumption: the matmul function has been replaced by an explicit expand of the weight matrix and bmm leading to 30% less memory consumption. However, indexedconv still uses way more memory than torch.nn.Conv2d.