options.register("batchSize", 1, VarParsing.multiplicity.singleton,
VarParsing.varType.int)
options.register("producer", "TritonImageProducer",
VarParsing.multiplicity.singleton, VarParsing.varType.string)
options.register("modelName", "resnet50_netdef",
VarParsing.multiplicity.singleton, VarParsing.varType.string)
options.register("mode", "PseudoAsync", VarParsing.multiplicity.singleton,
VarParsing.varType.string)
options.register("verbose", False, VarParsing.multiplicity.singleton,
VarParsing.varType.bool)
options.parseArguments()
if len(options.params) > 0:
with open(options.params, 'r') as pfile:
pdict = json.load(pfile)
options.address = pdict["address"]
options.port = int(pdict["port"])
print("server = " + options.address + ":" + str(options.port))
# check producer/model
models = {
"TritonImageProducer": "resnet50_netdef",
"TritonGraphProducer": "gat_test",
}
if options.producer not in models:
raise ValueError("Unknown producer: " + options.producer)
process = cms.Process('tritonTest')
process.maxEvents = cms.untracked.PSet(
