/diagonalwise-refactorization-caffe

Diagonalwise Refactorization: An Efficient Training Method for Depthwise Convolutions (in Caffe)

Primary LanguageC++OtherNOASSERTION

Diagonalwise Refactorization: An Efficient Training Method for Depthwise Convolutions

This is the Caffe implementation of Diagonalwise Refactorization (Forked from Caffe).

Diagonalwise Refactorization is an efficient implementation for depthwise convolutions. The key idea of Diagonalwise Refactorization is to rearrange the weight vectors of a depthwise convolution into a large diagonal weight matrixi, so as to convert the depthwise convolution into one single standard convolution, which is well supported by the cuDNN library that is highly-optimized for GPU computations.

TODO:

  1. Loading weights from Proto.
  2. Saving / Loading weights from HDF5.