fixed init_net, predict_net path

src/main/resources/templates/caffe2/CNNPredictor.ftl

@@ -18,8 +18,8 @@
 #include <iostream>
 #include <map>
 
-CAFFE2_DEFINE_string(init_net_${tc.fileNameWithoutEnding}, "./model/${tc.componentName}/init_net_${tc.fileNameWithoutEnding}.pb", "The given path to the init protobuffer.");
-CAFFE2_DEFINE_string(predict_net_${tc.fileNameWithoutEnding}, "./model/${tc.componentName}/predict_net_${tc.fileNameWithoutEnding}.pb", "The given path to the predict protobuffer.");
+CAFFE2_DEFINE_string(init_net_${tc.fileNameWithoutEnding}, "./model/${tc.componentName}/init_net.pb", "The given path to the init protobuffer.");
+CAFFE2_DEFINE_string(predict_net_${tc.fileNameWithoutEnding}, "./model/${tc.componentName}/predict_net.pb", "The given path to the predict protobuffer.");
 
 using namespace caffe2;
 

src/test/resources/target_code/CNNPredictor_LeNet.h

@@ -18,8 +18,8 @@
 #include <iostream>
 #include <map>
 
-CAFFE2_DEFINE_string(init_net_CNNPredictor_LeNet, "./model/LeNet/init_net_CNNPredictor_LeNet.pb", "The given path to the init protobuffer.");
-CAFFE2_DEFINE_string(predict_net_CNNPredictor_LeNet, "./model/LeNet/predict_net_CNNPredictor_LeNet.pb", "The given path to the predict protobuffer.");
+CAFFE2_DEFINE_string(init_net_CNNPredictor_LeNet, "./model/LeNet/init_net.pb", "The given path to the init protobuffer.");
+CAFFE2_DEFINE_string(predict_net_CNNPredictor_LeNet, "./model/LeNet/predict_net.pb", "The given path to the predict protobuffer.");
 
 using namespace caffe2;
 

src/test/resources/target_code/CNNPredictor_VGG16.h

@@ -18,8 +18,8 @@
 #include <iostream>
 #include <map>
 
-CAFFE2_DEFINE_string(init_net_CNNPredictor_VGG16, "./model/VGG16/init_net_CNNPredictor_VGG16.pb", "The given path to the init protobuffer.");
-CAFFE2_DEFINE_string(predict_net_CNNPredictor_VGG16, "./model/VGG16/predict_net_CNNPredictor_VGG16.pb", "The given path to the predict protobuffer.");
+CAFFE2_DEFINE_string(init_net_CNNPredictor_VGG16, "./model/VGG16/init_net.pb", "The given path to the init protobuffer.");
+CAFFE2_DEFINE_string(predict_net_CNNPredictor_VGG16, "./model/VGG16/predict_net.pb", "The given path to the predict protobuffer.");
 
 using namespace caffe2;