some cleanup

diff --git a/src/caffe/net.cpp b/src/caffe/net.cpp
index fbb109f..e976dfd 100644 (file)
--- a/src/caffe/net.cpp
+++ b/src/caffe/net.cpp
@@ -1,6 +1,5 @@
 // Copyright Yangqing Jia 2013
 
-#include <fstream>
 #include <map>
 #include <set>
 #include <string>

diff --git a/src/caffe/test/test_split_layer.cpp b/src/caffe/test/test_split_layer.cpp
index aefa50c..19df34f 100644 (file)
--- a/src/caffe/test/test_split_layer.cpp
+++ b/src/caffe/test/test_split_layer.cpp
@@ -123,8 +123,6 @@ class SplitLayerInsertionTest : public ::testing::Test {
         output_param_string, &expected_output_param));
     NetParameter actual_output_param;
     insert_splits(input_param, &actual_output_param);
-    CHECK_EQ(expected_output_param.DebugString(),
-        actual_output_param.DebugString());
     EXPECT_EQ(expected_output_param.DebugString(),
         actual_output_param.DebugString());
   }

diff --git a/src/caffe/util/insert_splits.cpp b/src/caffe/util/insert_splits.cpp
index df8bf5d..c963dc8 100644 (file)
--- a/src/caffe/util/insert_splits.cpp
+++ b/src/caffe/util/insert_splits.cpp
@@ -36,6 +36,8 @@ void insert_splits(const NetParameter& param, NetParameter* param_split) {
       blob_name_to_bottom_split_idx[blob_name] = 0;
     }
   }
+  // Create split layer for any input blobs user by other layers as bottom
+  // blobs more than once.
   for (int i = 0; i < param.input_size(); ++i) {
     const string& blob_name = param.input(i);
     const int split_count = blob_name_to_bottom_count[blob_name];
@@ -61,7 +63,7 @@ void insert_splits(const NetParameter& param, NetParameter* param_split) {
         layer_connection->set_bottom(j, split_blob_name);
       }
     }
-    // Create split blob for any top blobs used by other layers as bottom
+    // Create split layer for any top blobs used by other layers as bottom
     // blobs more than once.
     for (int j = 0; j < layer_connection->top_size(); ++j) {
       const string& blob_name = layer_connection->top(j);