ONNXRuntime/test/test_particlenet_from_miniaod_ak8_cfg.py

0001 ### CMSSW command line parameter parser
0002 import FWCore.ParameterSet.Config as cms
0003
0004
0005 process = cms.Process("test")
0006
0007 # Message Logger settings
0008 process.load("FWCore.MessageService.MessageLogger_cfi")
0009 process.MessageLogger.cerr.FwkReport.reportEvery = 250
0010 process.load("RecoBTag.ONNXRuntime.pfParticleNetFromMiniAODAK8_cff")
0011
0012 process.maxEvents = cms.untracked.PSet(
0013     input = cms.untracked.int32(100)
0014 )
0015
0016 process.options = cms.untracked.PSet(
0017     allowUnscheduled = cms.untracked.bool(True),
0018     wantSummary      = cms.untracked.bool(True),
0019     numberOfThreads  = cms.untracked.uint32(8),
0020     numberOfStreams = cms.untracked.uint32(0)
0021 )
0022
0023 process.source = cms.Source("PoolSource",
0024     fileNames = cms.untracked.vstring("/store/mc/RunIISummer20UL18MiniAODv2/TTToSemiLeptonic_TuneCP5_13TeV-powheg-pythia8/MINIAODSIM/106X_upgrade2018_realistic_v16_L1v1-v2/130000/25BF763A-BF41-E242-86A2-5E0BE8EF605C.root")
0025 )
0026
0027 process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
0028 process.load('Configuration.StandardSequences.GeometryDB_cff')
0029 process.load('Configuration.StandardSequences.MagneticField_cff')
0030 process.load('TrackingTools.TransientTrack.TransientTrackBuilder_cfi')
0031
0032 from Configuration.AlCa.GlobalTag import GlobalTag
0033 process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:phase1_2018_realistic', '')
0034
0035
0036 from PhysicsTools.PatAlgos.producersLayer1.jetUpdater_cfi import updatedPatJets
0037 process.slimmedJetsUpdated = updatedPatJets.clone(
0038     jetSource = "slimmedJetsAK8",
0039     addJetCorrFactors = False,
0040     discriminatorSources = cms.VInputTag(
0041         cms.InputTag("pfParticleNetFromMiniAODAK8JetTags:probHtt"),
0042         cms.InputTag("pfParticleNetFromMiniAODAK8JetTags:probHtm"),
0043         cms.InputTag("pfParticleNetFromMiniAODAK8JetTags:probHte"),
0044         cms.InputTag("pfParticleNetFromMiniAODAK8JetTags:probHbb"),
0045         cms.InputTag("pfParticleNetFromMiniAODAK8JetTags:probHcc"),
0046         cms.InputTag("pfParticleNetFromMiniAODAK8JetTags:probHqq"),
0047         cms.InputTag("pfParticleNetFromMiniAODAK8JetTags:probHgg"),
0048         cms.InputTag("pfParticleNetFromMiniAODAK8JetTags:probQCD2hf"),
0049         cms.InputTag("pfParticleNetFromMiniAODAK8JetTags:probQCD1hf"),
0050         cms.InputTag("pfParticleNetFromMiniAODAK8JetTags:probQCD0hf"),
0051         cms.InputTag("pfParticleNetFromMiniAODAK8JetTags:masscorr"),
0052     )
0053 )
0054
0055
0056 process.path = cms.Path(
0057     process.slimmedJetsUpdated,process.pfParticleNetFromMiniAODAK8Task
0058 )
0059
0060 process.output = cms.OutputModule( "PoolOutputModule",
0061     fileName = cms.untracked.string( "test.root" ),
0062     compressionAlgorithm = cms.untracked.string('LZMA'),
0063     compressionLevel = cms.untracked.int32(4),
0064     eventAutoFlushCompressedSize = cms.untracked.int32(31457280),
0065     dataset = cms.untracked.PSet(
0066         dataTier = cms.untracked.string( 'RECO' ),
0067         filterName = cms.untracked.string( '' )
0068     ),
0069     overrideBranchesSplitLevel = cms.untracked.VPSet(),
0070     outputCommands = cms.untracked.vstring(
0071         'drop *',
0072         'keep *_*slimmedJetsUpdated*_*_*',
0073     )
0074 )
0075
0076 process.endpath = cms.EndPath(process.output);
0077 process.schedule = cms.Schedule(process.path,process.endpath);
0078