Corner case fix

diff --git a/src/caffe/layers/split_layer.cpp b/src/caffe/layers/split_layer.cpp
index ca3e6a257eac3f1c7fe8331e01bc04f1f1ad848d..a204b781ef12070ae18b78871772ad781653c515 100644 (file)
--- a/src/caffe/layers/split_layer.cpp
+++ b/src/caffe/layers/split_layer.cpp
@@ -16,7 +16,7 @@ void SplitLayer<Ftype, Btype>::Reshape(const vector<Blob*>& bottom, const vector
     // some strange effects in practice...)
     CHECK_NE(top[i], bottom[0]) << "SplitLayer " << this->type()
         << " does not allow in-place computation.";
-    if(top[i]->count() != bottom[0]->count()) {
+    if (top[i]->count() != bottom[0]->count()) {
       top[i]->ReshapeLike(*bottom[0]);
     }
     top[i]->ShareData(*bottom[0]);