diff --git a/PhysicsTools/NanoAOD/plugins/BJetEnergyRegressionMVA.cc b/PhysicsTools/NanoAOD/plugins/BJetEnergyRegressionMVA.cc index 8ab46a5a552d6..c36922c3c5f99 100644 --- a/PhysicsTools/NanoAOD/plugins/BJetEnergyRegressionMVA.cc +++ b/PhysicsTools/NanoAOD/plugins/BJetEnergyRegressionMVA.cc @@ -25,7 +25,7 @@ #include "RecoVertex/VertexPrimitives/interface/ConvertToFromReco.h" #include "RecoVertex/VertexPrimitives/interface/VertexState.h" -#include "PhysicsTools/PatAlgos/plugins/BaseMVAValueMapProducer.h" +#include "PhysicsTools/PatAlgos/interface/BaseMVAValueMapProducer.h" #include class BJetEnergyRegressionMVA : public BaseMVAValueMapProducer { diff --git a/PhysicsTools/NanoAOD/python/jets_cff.py b/PhysicsTools/NanoAOD/python/jets_cff.py index 5d51ad4cf93c5..35cb0a05235f3 100644 --- a/PhysicsTools/NanoAOD/python/jets_cff.py +++ b/PhysicsTools/NanoAOD/python/jets_cff.py @@ -537,6 +537,7 @@ from Configuration.Eras.Modifier_fastSim_cff import fastSim def nanoAOD_refineFastSim_bTagDeepFlav(process): + fastSim.toModify( process.jetTable.variables, btagDeepFlavBunrefined = process.jetTable.variables.btagDeepFlavB.clone(), btagDeepFlavCvBunrefined = process.jetTable.variables.btagDeepFlavCvB.clone(), @@ -565,7 +566,7 @@ def nanoAOD_refineFastSim_bTagDeepFlav(process): src = cms.InputTag("linkedObjects","jets"), - weightFile=cms.FileInPath("PhysicsTools/NanoAOD/data/btagDeepFlavRefineNN_CHS.onnx"), + weightFile=cms.FileInPath("PhysicsTools/NanoAOD/data/btagDeepFlavRefineNN_CHS.onnx"),#local hack for backporting: cp /cvmfs/cms.cern.ch/slc7_amd64_gcc11/cms/data-PhysicsTools-NanoAOD/V01-03-00/PhysicsTools/NanoAOD/data/btagDeepFlavRefineNN_CHS.onnx PhysicsTools/NanoAOD/data/btagDeepFlavRefineNN_CHS.onnx # cp /nfs/dust/cms/user/wolfmor/Refinement/TrainingOutput/regression_model_20221127_opset11.onnx PhysicsTools/NanoAOD/data/btagDeepFlavRefineNN_CHS.onnx name = cms.string("btagDeepFlavRefineNN"), isClassifier = cms.bool(False), @@ -586,7 +587,7 @@ def nanoAOD_refineFastSim_bTagDeepFlav(process): outputFormulas = cms.vstring(["at(0)","at(1)","at(2)","at(3)"]), ) - fastSim.toModify(process.jetTablesTask, process.jetTablesTask.add(process.btagDeepFlavRefineNN)) + fastSim.toModify(process.jetTables, process.jetTables.insert(0,process.btagDeepFlavRefineNN)) return process diff --git a/PhysicsTools/PatAlgos/interface/BaseMVAValueMapProducer.h b/PhysicsTools/PatAlgos/interface/BaseMVAValueMapProducer.h index 07f94dbc677fa..6b7e064d5b837 100644 --- a/PhysicsTools/PatAlgos/interface/BaseMVAValueMapProducer.h +++ b/PhysicsTools/PatAlgos/interface/BaseMVAValueMapProducer.h @@ -278,6 +278,7 @@ void BaseMVAValueMapProducer::produce(edm::Event& iEvent, const edm::EventSet } } + template edm::ParameterSetDescription BaseMVAValueMapProducer::getDescription() { edm::ParameterSetDescription desc; @@ -299,6 +300,7 @@ edm::ParameterSetDescription BaseMVAValueMapProducer::getDescription() { desc.add("nThreads", 1)->setComment("number of threads"); desc.add("singleThreadPool", "no_threads"); desc.add("batch_eval", false)->setComment("Run inference in batch instead of per-object"); + desc.add("disableONNXGraphOpt", false)->setComment("Disable ONNX runtime graph optimization"); return desc; } diff --git a/PhysicsTools/PatAlgos/plugins/BaseMVAValueMapProducer.cc b/PhysicsTools/PatAlgos/plugins/BaseMVAValueMapProducer.cc index bcb00b910bc6d..42935571a28ac 100644 --- a/PhysicsTools/PatAlgos/plugins/BaseMVAValueMapProducer.cc +++ b/PhysicsTools/PatAlgos/plugins/BaseMVAValueMapProducer.cc @@ -2,7 +2,7 @@ // // Package: PhysicsTools/PatAlgos // Class: BaseMVAValueMapProducer -// +// /**\class BaseMVAValueMapProducer BaseMVAValueMapProducer.cc PhysicsTools/PatAlgos/plugins/BaseMVAValueMapProducer.cc Description: [one line class summary] @@ -26,3 +26,4 @@ typedef BaseMVAValueMapProducer EleBaseMVAValueMapProducer; DEFINE_FWK_MODULE(JetBaseMVAValueMapProducer); DEFINE_FWK_MODULE(MuonBaseMVAValueMapProducer); DEFINE_FWK_MODULE(EleBaseMVAValueMapProducer); +