HLTrigger/Configuration/python/HLT_75e33/modules/hltInitialStepTrackTorchClassifierOutput_cfi.py - master - githubmirror/cmssw

HLTrigger/Configuration/python/HLT_75e33/modules/hltInitialStepTrackTorchClassifierOutput_cfi.py

14 строк621 B

aashayarora

Add pytorch model for phase-2 hlt track classification

09 июл 2026, 20:53

09 июл 2026, 20:5322195b2

import FWCore.ParameterSet.Config as cms

hltInitialStepTrackTorchClassifierOutput = cms.EDProducer("TrackTorchClassifierFromSoA",
    src = cms.InputTag("hltInitialStepTracks"),
    scores = cms.InputTag("hltInitialStepTrackTorchClassifier"),
    features = cms.InputTag("hltInitialStepTrackFeatureExtractor"),
    copyTrajectories = cms.bool(False),
    minScore = cms.double(0.377),
    dxyThreshold = cms.double(0.5),
    highDxyMinScore = cms.double(0.267)
)

from Configuration.ProcessModifiers.mtd_at_hlt_cff import mtd_at_hlt
mtd_at_hlt.toModify(hltInitialStepTrackTorchClassifierOutput, copyTrajectories = True)