Skip to content

Commit d4cf6b8

Browse files
committed
Add caching and swap to torch==2.8.0+cpu
1 parent 33f4c60 commit d4cf6b8

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

benchmarks/lowq2_reconstruction/Snakefile

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ rule filter_hepmc_for_training:
1313
script=workflow.source_path("filterHEPMC3.py"),
1414
output:
1515
SIMOUTDIR+"Low-Q2_Training_Events.hepmc3.tree.root",
16+
cache: True
1617
params:
1718
events="root://dtn-eic.jlab.org//volatile/eic/EPIC/EVGEN/SIDIS/pythia6-eic/1.0.0/18x275/q2_0to1/pythia_ep_noradcor_18x275_q2_0.000000001_1.0_run1.ab.hepmc3.tree.root",
1819
shell:
@@ -27,6 +28,7 @@ rule lowq2_training_sim:
2728
events=SIMOUTDIR+"Low-Q2_Training_Events.hepmc3.tree.root",
2829
output:
2930
SIMOUTDIR+"Low-Q2_Training_SimEvents_{CAMPAIGN}.edm4hep.root",
31+
cache: True
3032
shell:
3133
"""
3234
exec npsim \

benchmarks/lowq2_reconstruction/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
--index-url https://download.pytorch.org/whl/cpu
22
numpy==2.3.1
33
#gpu version of torch on eicweb doesn't speed up things for our small model and takes a long time to install
4-
torch==2.5.1
4+
torch==2.8.0+cpu
55
uproot==5.6.3
66
pyHepMC3==3.3.0
77
onnx==1.17.0

0 commit comments

Comments
 (0)