diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index dbcc78e12dc8e19fc02e6baf1fa6e8176701eac9..3c94364965749e8867b0de875c30afb2e96b8397 100644 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -61,6 +61,7 @@ include: - local: 'benchmarks/clustering/config.yml' - local: 'benchmarks/rich/config.yml' - local: 'benchmarks/imaging_ecal/config.yml' + - local: 'benchmarks/imaging_shower_ML/config.yml' final_report: diff --git a/benchmarks/imaging_shower_ML/config.yml b/benchmarks/imaging_shower_ML/config.yml index 70c79ecdd46342bb19577230227789ff0d6d94ea..bccac7e7491b8c022c847603c071ac366d598fb9 100644 --- a/benchmarks/imaging_shower_ML/config.yml +++ b/benchmarks/imaging_shower_ML/config.yml @@ -1,12 +1,31 @@ -ml_shower:tagging : +ml_shower:tagging_epi : extends: .rec_benchmark + when: manual + stage: process + timeout: 8 hour + script: + - python3 benchmarks/imaging_shower_ML/sim_rec_tag.py -t imshower -n 10000 --particles "electron,pion-,pion-" --pmin 0.5 --pmax 10 + + +ml_shower:tagging_e : + extends: .rec_benchmark + when: manual stage: process timeout: 8 hour script: - python3 benchmarks/imaging_shower_ML/sim_rec_tag.py -t imshower -n 10000 --particles "electron" --pmin 0.5 --pmax 10 +ml_shower:tagging_pi : + extends: .rec_benchmark + when: manual + stage: process + timeout: 8 hour + script: + - python3 benchmarks/imaging_shower_ML/sim_rec_tag.py -t imshower -n 10000 --particles "pion-" --pmin 0.5 --pmax 10 + ml_shower:training: extends: .rec_benchmark + when: manual stage: process needs: ["ml_shower:tagging"] script: @@ -15,8 +34,9 @@ ml_shower:training: ml_shower:test: extends: .rec_benchmark + when: manual stage: collect - needs: ["ml_shower:tagging"] + needs: ["ml_shower:training"] script: - ls -lrth # TODO