fix memory bug in DNN-based ID

cms-sw · Nov 4, 2021 · e1dd9c4 · e1dd9c4
1 parent d1de9af
commit e1dd9c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/RecoEgamma/EgammaElectronProducers/plugins/GsfElectronProducer.cc b/RecoEgamma/EgammaElectronProducers/plugins/GsfElectronProducer.cc
@@ -291,7 +291,7 @@ void GsfElectronProducer::fillDescriptions(edm::ConfigurationDescriptions& descr
     psd1.add<bool>("enabled", false);
     psd1.add<std::string>("inputTensorName", "FirstLayer_input");
     psd1.add<std::string>("outputTensorName", "sequential/FinalLayer/Softmax");
-    psd1.add<uint>("outputDim", 3);  // Dimension of output vector
+    psd1.add<uint>("outputDim", 5);  // Number of output nodes of DNN
     psd1.add<std::vector<std::string>>(
         "modelsFiles",
         {"RecoEgamma/ElectronIdentification/data/Ele_PFID_dnn/lowpT/lowpT_modelDNN.pb",