Add new DeepFlavour training

emilbols · Jul 6, 2018 · 94ceae2 · 94ceae2
1 parent 47e7c01
commit 94ceae2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/RecoBTag/TensorFlow/plugins/DeepFlavourTFJetTagsProducer.cc b/RecoBTag/TensorFlow/plugins/DeepFlavourTFJetTagsProducer.cc
@@ -136,9 +136,9 @@ void DeepFlavourTFJetTagsProducer::fillDescriptions(edm::ConfigurationDescriptio
   desc.add<std::vector<std::string>>("input_names", 
     { "input_1", "input_2", "input_3", "input_4", "input_5" });
   desc.add<edm::FileInPath>("graph_path",
-    edm::FileInPath("RecoBTag/Combined/data/DeepFlavourV01_GraphDef_PtCut/constant_graph.pb"));
+    edm::FileInPath("RecoBTag/Combined/data/DeepFlavourV03_10X_training/constant_graph.pb"));
   desc.add<std::vector<std::string>>("lp_names",
-    { "globals_input_batchnorm/keras_learning_phase" });
+    { "cpf_input_batchnorm/keras_learning_phase" });
   desc.add<std::vector<std::string>>("output_names",
     { "ID_pred/Softmax", "regression_pred/BiasAdd" });
   {