Update main.cpp

koosyong · koosyong · commit f4f183aa0dd7 · 2016-01-08T09:54:41.000+01:00
diff --git a/examples/ex4_layer/main.cpp b/examples/ex4_layer/main.cpp
@@ -4,6 +4,7 @@
 #include "caffe/blob.hpp"
 #include "caffe/common.hpp"
 #include "caffe/filler.hpp"
+#include "caffe/layers/argmax_layer.hpp"
 
 using namespace caffe;
 using namespace std;
