Skip to content

qinzheng93/diagonalwise-refactorization-caffe

 
 

Repository files navigation

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.

About

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

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • C++ 80.5%
  • Python 8.5%
  • Cuda 6.2%
  • CMake 2.8%
  • MATLAB 0.9%
  • Makefile 0.7%
  • Shell 0.4%