Commit 8f10cc34 authored by Manuel Guth's avatar Manuel Guth
Browse files

Merge branch 'fixing_lwtnn_conversion' into 'master'

Fixing small issues with convert lwtnn for DIPS

See merge request atlas-flavor-tagging-tools/algorithms/umami!93
parents 612d30e5 d2dfd48c
# Add train_config filepath
train_config: /work/ws/nemo/fr_af1100-Training-Simulations-0/b-Tagging/packages/umami/examples/Dips-PFlow-Training-config.yaml
train_config: /work/ws/nemo/fr_af1100-Training-Simulations-0/b-Tagging/Submission_Scripts/Train_Dips_Loose/configs/Dips-PFlow-Training-config.yaml
# Add preprocess_config with which the used model is trained
preprocess_config: /work/ws/nemo/fr_af1100-Training-Simulations-0/b-Tagging/packages/umami/examples/PFlow-Preprocessing.yaml
preprocess_config: /work/ws/nemo/fr_af1100-Training-Simulations-0/b-Tagging/Submission_Scripts/Train_Dips_Loose/configs/PFlow-Preprocessing.yaml
# Path to Variable dict used in preprocessing
var_dict: /work/ws/nemo/fr_af1100-Training-Simulations-0/b-Tagging/packages/umami/umami/configs/Dips_Variables.yaml
var_dict: /work/ws/nemo/fr_af1100-Training-Simulations-0/b-Tagging/Submission_Scripts/Train_Dips_Loose/configs/Dips_Variables.yaml
# File to test the dips integration in the dumper
test_file: /work/ws/nemo/fr_af1100-Training-Simulations-0/test/
test_file: /work/ws/nemo/fr_af1100-Training-Simulations-0/DAOD_PHYSVAL.25394913._000001.pool.root.1_loose.h5
# Path to model dirs
model_file: /work/ws/nemo/fr_af1100-Training-Simulations-0/b-Tagging/packages/umami/umami/dips_Loose_lr_0.001_bs_15000_epoch_200_nTrainJets_Full/dips_model_59.h5
\ No newline at end of file
......@@ -121,9 +121,9 @@ def __run():
if "dips" in args.tagger.lower():
with open(args.scale_dict, "r") as f:
scale_dict = json.load(f)["tracks"]
scale_dict = json.load(f)
track_variables = GetJetVariables(scale_dict, variable_config)
track_variables = GetTrackVariables(scale_dict, variable_config)
print("Found %i variables" % len(track_variables))
inputs = {}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment