qinzheng93 / diagonalwise-refactorization-caffe

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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)

License:Other


Languages

Language:C++ 80.6%Language:Python 8.5%Language:Cuda 6.2%Language:CMake 2.8%Language:MATLAB 0.9%Language:Makefile 0.7%Language:Shell 0.4%