squash! Add pruning possibilities at inner_product_layer

BVLC · Jun 12, 2016 · b5ed779 · b5ed779
1 parent 81c9127
commit b5ed779
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/include/caffe/layers/inner_product_layer.hpp b/include/caffe/layers/inner_product_layer.hpp
@@ -44,7 +44,8 @@ class InnerProductLayer : public Layer<Dtype> {
   int N_;
   bool bias_term_;
   bool pruned_;
-  vector<shared_ptr<Blob<Dtype> > > masks_;  // contain masks for deep compression
+  // contain masks for deep compression
+  vector<shared_ptr<Blob<Dtype> > > masks_;
   Dtype pruning_coeff_;  // pruning rate for deep compression
   Blob<Dtype> bias_multiplier_;
   bool transpose_;  ///< if true, assume transposed weights

diff --git a/src/caffe/util/math_functions.cpp b/src/caffe/util/math_functions.cpp
@@ -1,11 +1,11 @@
 #include <boost/math/special_functions/next.hpp>
 #include <boost/random.hpp>
 
-#include <limits>
 #include <cmath>
+#include <functional>
+#include <limits>
 #include <utility>
 #include <vector>
-#include <functional>
 
 #include "caffe/common.hpp"
 #include "caffe/util/math_functions.hpp"