diff --git a/HiggsInteractionNet/V00/modelfile b/HiggsInteractionNet/V00/modelfile new file mode 120000 index 0000000..dfe1cd6 --- /dev/null +++ b/HiggsInteractionNet/V00/modelfile @@ -0,0 +1 @@ +../../models/higgsInteractionNet/1 \ No newline at end of file diff --git a/ParticleNetFromMiniAODAK4/CHS/Central/modelfile b/ParticleNetFromMiniAODAK4/CHS/Central/modelfile new file mode 120000 index 0000000..758259d --- /dev/null +++ b/ParticleNetFromMiniAODAK4/CHS/Central/modelfile @@ -0,0 +1 @@ +../../../models/particleNetFromMiniAODAK4CHSCentral/1 \ No newline at end of file diff --git a/ParticleNetFromMiniAODAK4/CHS/Forward/modelfile b/ParticleNetFromMiniAODAK4/CHS/Forward/modelfile new file mode 120000 index 0000000..f17a459 --- /dev/null +++ b/ParticleNetFromMiniAODAK4/CHS/Forward/modelfile @@ -0,0 +1 @@ +../../../models/particleNetFromMiniAODAK4CHSForward/1 \ No newline at end of file diff --git a/ParticleNetFromMiniAODAK4/PUPPI/Central/modelfile b/ParticleNetFromMiniAODAK4/PUPPI/Central/modelfile new file mode 120000 index 0000000..fbca89c --- /dev/null +++ b/ParticleNetFromMiniAODAK4/PUPPI/Central/modelfile @@ -0,0 +1 @@ +../../../models/particleNetFromMiniAODAK4PuppiCentral/1 \ No newline at end of file diff --git a/ParticleNetFromMiniAODAK4/PUPPI/Forward/modelfile b/ParticleNetFromMiniAODAK4/PUPPI/Forward/modelfile new file mode 120000 index 0000000..adda797 --- /dev/null +++ b/ParticleNetFromMiniAODAK4/PUPPI/Forward/modelfile @@ -0,0 +1 @@ +../../../models/particleNetFromMiniAODAK4PuppiForward/1 \ No newline at end of file diff --git a/ParticleNetFromMiniAODAK8/modelfile b/ParticleNetFromMiniAODAK8/modelfile new file mode 120000 index 0000000..796fb38 --- /dev/null +++ b/ParticleNetFromMiniAODAK8/modelfile @@ -0,0 +1 @@ +../models/particleNetFromMiniAODAK8/1 \ No newline at end of file diff --git a/HiggsInteractionNet/V00/IN.onnx b/models/higgsInteractionNet/1/model.onnx similarity index 100% rename from HiggsInteractionNet/V00/IN.onnx rename to models/higgsInteractionNet/1/model.onnx diff --git a/models/higgsInteractionNet/config.pbtxt b/models/higgsInteractionNet/config.pbtxt new file mode 100644 index 0000000..a6de55f --- /dev/null +++ b/models/higgsInteractionNet/config.pbtxt @@ -0,0 +1,27 @@ +name: "higgsInteractionNet" +platform: "onnxruntime_onnx" +max_batch_size : 500 +dynamic_batching { + preferred_batch_size: [ 200 ] +} +input [ + { + name: "input_cpf" + data_type: TYPE_FP32 + dims: [ 30, 60 ] + }, + { + name: "input_sv" + data_type: TYPE_FP32 + dims: [ 14, 5 ] + } +] +output [ + { + name: "output1" + data_type: TYPE_FP32 + dims: [ 2 ] + label_filename: "higgs_interactionnet_labels.txt" + } +] +optimization {graph : {level : -1}} diff --git a/models/higgsInteractionNet/higgs_interactionnet_labels.txt b/models/higgsInteractionNet/higgs_interactionnet_labels.txt new file mode 100644 index 0000000..eda7cb1 --- /dev/null +++ b/models/higgsInteractionNet/higgs_interactionnet_labels.txt @@ -0,0 +1,2 @@ +QCD +Hbb diff --git a/models/higgsInteractionNet/preprocess.json b/models/higgsInteractionNet/preprocess.json new file mode 100644 index 0000000..74dbe98 --- /dev/null +++ b/models/higgsInteractionNet/preprocess.json @@ -0,0 +1,422 @@ +{ + "output_names": [ + "QCD", + "Hbb" + ], + "input_names": [ + "input_cpf", + "input_sv" + ], + "input_cpf": { + "var_names": [ + "pfcand_ptrel", + "pfcand_erel", + "pfcand_phirel", + "pfcand_etarel", + "pfcand_deltaR", + "pfcand_drminsv", + "pfcand_drsubjet1", + "pfcand_drsubjet2", + "pfcand_dz", + "pfcand_dzsig", + "pfcand_dxy", + "pfcand_dxysig", + "pfcand_normchi2", + "pfcand_quality", + "pfcand_dptdpt", + "pfcand_detadeta", + "pfcand_dphidphi", + "pfcand_dxydxy", + "pfcand_dzdz", + "pfcand_dxydz", + "pfcand_dphidxy", + "pfcand_dlambdadz", + "pfcand_btagEtaRel", + "pfcand_btagPtRatio", + "pfcand_btagPParRatio", + "pfcand_btagSip2dVal", + "pfcand_btagSip2dSig", + "pfcand_btagSip3dVal", + "pfcand_btagSip3dSig", + "pfcand_btagJetDistVal" + ], + "var_infos": { + "pfcand_ptrel": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_erel": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_phirel": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_etarel": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_deltaR": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_drminsv": { + "median": 0.8, + "norm_factor": 1.0, + "replace_inf_value": 0.8, + "lower_bound": -0.8, + "upper_bound": 0.0, + "pad": 0 + }, + "pfcand_drsubjet1": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_drsubjet2": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_dz": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_dzsig": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -2000, + "upper_bound": 2000, + "pad": 0 + }, + "pfcand_dxy": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_dxysig": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -2000, + "upper_bound": 2000, + "pad": 0 + }, + "pfcand_normchi2": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 300.0, + "lower_bound": -1.0, + "upper_bound": 300, + "pad": 0 + }, + "pfcand_quality": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_dptdpt": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_detadeta": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_dphidphi": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_dxydxy": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_dzdz": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_dxydz": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_dphidxy": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_dlambdadz": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_btagEtaRel": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 15, + "pad": 0 + }, + "pfcand_btagPtRatio": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1, + "upper_bound": 10, + "pad": 0 + }, + "pfcand_btagPParRatio": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -10, + "upper_bound": 10, + "pad": 0 + }, + "pfcand_btagSip2dVal": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1, + "upper_bound": 70, + "pad": 0 + }, + "pfcand_btagSip2dSig": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1, + "upper_bound": 4e4, + "pad": 0 + }, + "pfcand_btagSip3dVal": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1, + "upper_bound": 1e5, + "pad": 0 + }, + "pfcand_btagSip3dSig": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -1, + "upper_bound": 4e4, + "pad": 0 + }, + "pfcand_btagJetDistVal": { + "median": 0.0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -20, + "upper_bound": 1, + "pad": 0 + } + }, + "min_length": 60, + "max_length": 60 + }, + "input_sv": { + "var_names": [ + "sv_ptrel", + "sv_erel", + "sv_phirel", + "sv_etarel", + "sv_deltaR", + "sv_pt", + "sv_mass", + "sv_ntracks", + "sv_normchi2", + "sv_dxy", + "sv_dxysig", + "sv_d3d", + "sv_d3dsig", + "sv_costhetasvpv" + ], + "var_infos": { + "sv_ptrel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "sv_erel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "sv_phirel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "sv_etarel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "sv_deltaR": { + "median": 0.5, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -2.0, + "upper_bound": 0.0, + "pad": 0 + }, + "sv_pt": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "sv_mass": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "sv_ntracks": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "sv_normchi2": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 1000, + "lower_bound": -1000, + "upper_bound": 1000, + "pad": 0 + }, + "sv_dxy": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "sv_dxysig": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1.0, + "upper_bound": 800.0, + "pad": 0 + }, + "sv_d3d": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "sv_d3dsig": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1.0, + "upper_bound": 800.0, + "pad": 0 + }, + "sv_costhetasvpv": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + } + }, + "min_length": 5, + "max_length": 5 + } +} diff --git a/ParticleNetFromMiniAODAK4/CHS/Central/particle-net.onnx b/models/particleNetFromMiniAODAK4CHSCentral/1/model.onnx similarity index 100% rename from ParticleNetFromMiniAODAK4/CHS/Central/particle-net.onnx rename to models/particleNetFromMiniAODAK4CHSCentral/1/model.onnx diff --git a/models/particleNetFromMiniAODAK4CHSCentral/config.pbtxt b/models/particleNetFromMiniAODAK4CHSCentral/config.pbtxt new file mode 100644 index 0000000..24ba730 --- /dev/null +++ b/models/particleNetFromMiniAODAK4CHSCentral/config.pbtxt @@ -0,0 +1,62 @@ +name: "particleNetFromMiniAODAK4CHSCentral" +platform: "onnxruntime_onnx" +max_batch_size : 500 +dynamic_batching { + preferred_batch_size: [ 200 ] +} +input [ + { + name: "pf_points" + data_type: TYPE_FP32 + dims: [ 2, -1 ] + }, + { + name: "pf_features" + data_type: TYPE_FP32 + dims: [ 41, -1 ] + }, + { + name: "pf_mask" + data_type: TYPE_FP32 + dims: [ 1, -1 ] + }, + { + name: "sv_points" + data_type: TYPE_FP32 + dims: [ 2, -1 ] + }, + { + name: "sv_features" + data_type: TYPE_FP32 + dims: [ 10, -1 ] + }, + { + name: "sv_mask" + data_type: TYPE_FP32 + dims: [ 1, -1 ] + }, + { + name: "lt_points" + data_type: TYPE_FP32 + dims: [ 2, -1 ] + }, + { + name: "lt_features" + data_type: TYPE_FP32 + dims: [ 19, -1 ] + }, + { + name: "lt_mask" + data_type: TYPE_FP32 + dims: [ 1, -1 ] + } +] +output [ + { + name: "output" + data_type: TYPE_FP32 + dims: [ 20 ] + label_filename: "particlenet_labels.txt" + } +] +optimization {graph : {level : -1}} diff --git a/models/particleNetFromMiniAODAK4CHSCentral/particlenet_labels.txt b/models/particleNetFromMiniAODAK4CHSCentral/particlenet_labels.txt new file mode 100644 index 0000000..245f2aa --- /dev/null +++ b/models/particleNetFromMiniAODAK4CHSCentral/particlenet_labels.txt @@ -0,0 +1,20 @@ +mu +el +taup_1h0p +taup_1h1p +taup_1h2p +taup_3h0p +taup_3h1p +taum_1h0p +taum_1h1p +taum_1h2p +taum_3h0p +taum_3h1p +b +c +uds +g +pt +pt_q16 +pt_q84 +pt_nu diff --git a/ParticleNetFromMiniAODAK4/CHS/Forward/particle-net.onnx b/models/particleNetFromMiniAODAK4CHSForward/1/model.onnx similarity index 100% rename from ParticleNetFromMiniAODAK4/CHS/Forward/particle-net.onnx rename to models/particleNetFromMiniAODAK4CHSForward/1/model.onnx diff --git a/models/particleNetFromMiniAODAK4CHSForward/config.pbtxt b/models/particleNetFromMiniAODAK4CHSForward/config.pbtxt new file mode 100644 index 0000000..5249f7a --- /dev/null +++ b/models/particleNetFromMiniAODAK4CHSForward/config.pbtxt @@ -0,0 +1,62 @@ +name: "particleNetFromMiniAODAK4CHSForward" +platform: "onnxruntime_onnx" +max_batch_size : 500 +dynamic_batching { + preferred_batch_size: [ 200 ] +} +input [ + { + name: "pf_points" + data_type: TYPE_FP32 + dims: [ 2, -1 ] + }, + { + name: "pf_features" + data_type: TYPE_FP32 + dims: [ 40, -1 ] + }, + { + name: "pf_mask" + data_type: TYPE_FP32 + dims: [ 1, -1 ] + }, + { + name: "sv_points" + data_type: TYPE_FP32 + dims: [ 2, -1 ] + }, + { + name: "sv_features" + data_type: TYPE_FP32 + dims: [ 10, -1 ] + }, + { + name: "sv_mask" + data_type: TYPE_FP32 + dims: [ 1, -1 ] + }, + { + name: "lt_points" + data_type: TYPE_FP32 + dims: [ 2, -1 ] + }, + { + name: "lt_features" + data_type: TYPE_FP32 + dims: [ 19, -1 ] + }, + { + name: "lt_mask" + data_type: TYPE_FP32 + dims: [ 1, -1 ] + } +] +output [ + { + name: "output" + data_type: TYPE_FP32 + dims: [ 6 ] + label_filename: "particlenet_labels.txt" + } +] +optimization {graph : {level : -1}} diff --git a/models/particleNetFromMiniAODAK4CHSForward/particlenet_labels.txt b/models/particleNetFromMiniAODAK4CHSForward/particlenet_labels.txt new file mode 100644 index 0000000..8e74cc9 --- /dev/null +++ b/models/particleNetFromMiniAODAK4CHSForward/particlenet_labels.txt @@ -0,0 +1,6 @@ +q +g +pt +pt_q16 +pt_q84 +pt_nu diff --git a/ParticleNetFromMiniAODAK4/PUPPI/Central/particle-net.onnx b/models/particleNetFromMiniAODAK4PuppiCentral/1/model.onnx similarity index 100% rename from ParticleNetFromMiniAODAK4/PUPPI/Central/particle-net.onnx rename to models/particleNetFromMiniAODAK4PuppiCentral/1/model.onnx diff --git a/models/particleNetFromMiniAODAK4PuppiCentral/config.pbtxt b/models/particleNetFromMiniAODAK4PuppiCentral/config.pbtxt new file mode 100644 index 0000000..4b08acf --- /dev/null +++ b/models/particleNetFromMiniAODAK4PuppiCentral/config.pbtxt @@ -0,0 +1,62 @@ +name: "particleNetFromMiniAODAK4PuppiCentral" +platform: "onnxruntime_onnx" +max_batch_size : 500 +dynamic_batching { + preferred_batch_size: [ 200 ] +} +input [ + { + name: "pf_points" + data_type: TYPE_FP32 + dims: [ 2, -1 ] + }, + { + name: "pf_features" + data_type: TYPE_FP32 + dims: [ 41, -1 ] + }, + { + name: "pf_mask" + data_type: TYPE_FP32 + dims: [ 1, -1 ] + }, + { + name: "sv_points" + data_type: TYPE_FP32 + dims: [ 2, -1 ] + }, + { + name: "sv_features" + data_type: TYPE_FP32 + dims: [ 10, -1 ] + }, + { + name: "sv_mask" + data_type: TYPE_FP32 + dims: [ 1, -1 ] + }, + { + name: "lt_points" + data_type: TYPE_FP32 + dims: [ 2, -1 ] + }, + { + name: "lt_features" + data_type: TYPE_FP32 + dims: [ 19, -1 ] + }, + { + name: "lt_mask" + data_type: TYPE_FP32 + dims: [ 1, -1 ] + } +] +output [ + { + name: "output" + data_type: TYPE_FP32 + dims: [ 20 ] + label_filename: "particlenet_labels.txt" + } +] +optimization {graph : {level : -1}} diff --git a/models/particleNetFromMiniAODAK4PuppiCentral/particlenet_labels.txt b/models/particleNetFromMiniAODAK4PuppiCentral/particlenet_labels.txt new file mode 100644 index 0000000..245f2aa --- /dev/null +++ b/models/particleNetFromMiniAODAK4PuppiCentral/particlenet_labels.txt @@ -0,0 +1,20 @@ +mu +el +taup_1h0p +taup_1h1p +taup_1h2p +taup_3h0p +taup_3h1p +taum_1h0p +taum_1h1p +taum_1h2p +taum_3h0p +taum_3h1p +b +c +uds +g +pt +pt_q16 +pt_q84 +pt_nu diff --git a/ParticleNetFromMiniAODAK4/PUPPI/Forward/particle-net.onnx b/models/particleNetFromMiniAODAK4PuppiForward/1/model.onnx similarity index 100% rename from ParticleNetFromMiniAODAK4/PUPPI/Forward/particle-net.onnx rename to models/particleNetFromMiniAODAK4PuppiForward/1/model.onnx diff --git a/models/particleNetFromMiniAODAK4PuppiForward/config.pbtxt b/models/particleNetFromMiniAODAK4PuppiForward/config.pbtxt new file mode 100644 index 0000000..f2a8fe3 --- /dev/null +++ b/models/particleNetFromMiniAODAK4PuppiForward/config.pbtxt @@ -0,0 +1,62 @@ +name: "particleNetFromMiniAODAK4PuppiForward" +platform: "onnxruntime_onnx" +max_batch_size : 500 +dynamic_batching { + preferred_batch_size: [ 200 ] +} +input [ + { + name: "pf_points" + data_type: TYPE_FP32 + dims: [ 2, -1 ] + }, + { + name: "pf_features" + data_type: TYPE_FP32 + dims: [ 40, -1 ] + }, + { + name: "pf_mask" + data_type: TYPE_FP32 + dims: [ 1, -1 ] + }, + { + name: "sv_points" + data_type: TYPE_FP32 + dims: [ 2, -1 ] + }, + { + name: "sv_features" + data_type: TYPE_FP32 + dims: [ 10, -1 ] + }, + { + name: "sv_mask" + data_type: TYPE_FP32 + dims: [ 1, -1 ] + }, + { + name: "lt_points" + data_type: TYPE_FP32 + dims: [ 2, -1 ] + }, + { + name: "lt_features" + data_type: TYPE_FP32 + dims: [ 19, -1 ] + }, + { + name: "lt_mask" + data_type: TYPE_FP32 + dims: [ 1, -1 ] + } +] +output [ + { + name: "output" + data_type: TYPE_FP32 + dims: [ 6 ] + label_filename: "particlenet_labels.txt" + } +] +optimization {graph : {level : -1}} diff --git a/models/particleNetFromMiniAODAK4PuppiForward/particlenet_labels.txt b/models/particleNetFromMiniAODAK4PuppiForward/particlenet_labels.txt new file mode 100644 index 0000000..8e74cc9 --- /dev/null +++ b/models/particleNetFromMiniAODAK4PuppiForward/particlenet_labels.txt @@ -0,0 +1,6 @@ +q +g +pt +pt_q16 +pt_q84 +pt_nu diff --git a/ParticleNetFromMiniAODAK8/particle-net.onnx b/models/particleNetFromMiniAODAK8/1/model.onnx similarity index 100% rename from ParticleNetFromMiniAODAK8/particle-net.onnx rename to models/particleNetFromMiniAODAK8/1/model.onnx diff --git a/models/particleNetFromMiniAODAK8/config.pbtxt b/models/particleNetFromMiniAODAK8/config.pbtxt new file mode 100644 index 0000000..9ad1597 --- /dev/null +++ b/models/particleNetFromMiniAODAK8/config.pbtxt @@ -0,0 +1,47 @@ +name: "particleNetFromMiniAODAK8" +platform: "onnxruntime_onnx" +max_batch_size : 500 +dynamic_batching { + preferred_batch_size: [ 200 ] +} +input [ + { + name: "pf_points" + data_type: TYPE_FP32 + dims: [ 2, -1 ] + }, + { + name: "pf_features" + data_type: TYPE_FP32 + dims: [ 32, -1 ] + }, + { + name: "pf_mask" + data_type: TYPE_FP32 + dims: [ 1, -1 ] + }, + { + name: "sv_points" + data_type: TYPE_FP32 + dims: [ 2, -1 ] + }, + { + name: "sv_features" + data_type: TYPE_FP32 + dims: [ 10, -1 ] + }, + { + name: "sv_mask" + data_type: TYPE_FP32 + dims: [ 1, -1 ] + } +] +output [ + { + name: "output" + data_type: TYPE_FP32 + dims: [ 11 ] + label_filename: "particlenet_labels.txt" + } +] +optimization {graph : {level : -1}} diff --git a/models/particleNetFromMiniAODAK8/particlenet_labels.txt b/models/particleNetFromMiniAODAK8/particlenet_labels.txt new file mode 100644 index 0000000..ffa836a --- /dev/null +++ b/models/particleNetFromMiniAODAK8/particlenet_labels.txt @@ -0,0 +1,11 @@ +H_tt +H_tm +H_te +H_bb +H_cc +H_qq +H_gg +QCD_2hf +QCD_1hf +QCD_0hf +mass diff --git a/models/particlenet/config.pbtxt b/models/particlenet/config.pbtxt index aaf9558..848fa4a 100644 --- a/models/particlenet/config.pbtxt +++ b/models/particlenet/config.pbtxt @@ -43,4 +43,5 @@ output [ dims: [ 20 ] label_filename: "particlenet_labels.txt" } -] \ No newline at end of file +] +optimization {graph : {level : -1}} diff --git a/models/particlenet_AK4/config.pbtxt b/models/particlenet_AK4/config.pbtxt index 902b21d..ded4c0a 100644 --- a/models/particlenet_AK4/config.pbtxt +++ b/models/particlenet_AK4/config.pbtxt @@ -43,4 +43,5 @@ output [ dims: [ 8 ] label_filename: "particlenet_labels.txt" } -] \ No newline at end of file +] +optimization {graph : {level : -1}} diff --git a/models/particlenet_AK4_PT/config.pbtxt b/models/particlenet_AK4_PT/config.pbtxt index d93aad5..5e11764 100644 --- a/models/particlenet_AK4_PT/config.pbtxt +++ b/models/particlenet_AK4_PT/config.pbtxt @@ -43,4 +43,5 @@ output [ dims: [ 8 ] label_filename: "particlenet_labels.txt" } -] \ No newline at end of file +] +optimization {graph : {level : -1}} diff --git a/models/particlenet_AK8_MD-2prong/config.pbtxt b/models/particlenet_AK8_MD-2prong/config.pbtxt index 6058d38..bd6d6ee 100644 --- a/models/particlenet_AK8_MD-2prong/config.pbtxt +++ b/models/particlenet_AK8_MD-2prong/config.pbtxt @@ -43,4 +43,5 @@ output [ dims: [ 8 ] label_filename: "particlenet_labels.txt" } -] \ No newline at end of file +] +optimization {graph : {level : -1}} diff --git a/models/particlenet_AK8_MD-2prong_PT/config.pbtxt b/models/particlenet_AK8_MD-2prong_PT/config.pbtxt index 1b0de6c..8fd3051 100644 --- a/models/particlenet_AK8_MD-2prong_PT/config.pbtxt +++ b/models/particlenet_AK8_MD-2prong_PT/config.pbtxt @@ -43,4 +43,5 @@ output [ dims: [ 8 ] label_filename: "particlenet_labels.txt" } -] \ No newline at end of file +] +optimization {graph : {level : -1}} diff --git a/models/particlenet_AK8_MassRegression/config.pbtxt b/models/particlenet_AK8_MassRegression/config.pbtxt index 00487c0..02743fb 100644 --- a/models/particlenet_AK8_MassRegression/config.pbtxt +++ b/models/particlenet_AK8_MassRegression/config.pbtxt @@ -43,4 +43,5 @@ output [ dims: [ 1 ] label_filename: "particlenet_labels.txt" } -] \ No newline at end of file +] +optimization {graph : {level : -1}} diff --git a/models/particlenet_AK8_MassRegression_PT/config.pbtxt b/models/particlenet_AK8_MassRegression_PT/config.pbtxt index 93f8ee4..e0fa019 100644 --- a/models/particlenet_AK8_MassRegression_PT/config.pbtxt +++ b/models/particlenet_AK8_MassRegression_PT/config.pbtxt @@ -43,4 +43,5 @@ output [ dims: [ 1 ] label_filename: "particlenet_labels.txt" } -] \ No newline at end of file +] +optimization {graph : {level : -1}} diff --git a/models/particlenet_PT/config.pbtxt b/models/particlenet_PT/config.pbtxt index a096b33..0dfe6f5 100644 --- a/models/particlenet_PT/config.pbtxt +++ b/models/particlenet_PT/config.pbtxt @@ -43,4 +43,5 @@ output [ dims: [ 20 ] label_filename: "particlenet_labels.txt" } -] \ No newline at end of file +] +optimization {graph : {level : -1}}