diff --git a/src/Simulator.hpp b/src/Simulator.hpp index 867f8d7..a19ce28 100644 --- a/src/Simulator.hpp +++ b/src/Simulator.hpp @@ -10,6 +10,7 @@ #include "utils.hpp" #include "LayerData.hpp" +// Forward declare caffe::net so we don't have to include it here. namespace caffe { template