From 48a5e76bd492c2559bc76cbc005cbbb92d503060 Mon Sep 17 00:00:00 2001
From: Engin Eren <engin.eren@desy.de>
Date: Tue, 7 Jun 2022 11:01:36 +0200
Subject: [PATCH] switching to 50GeV only
---
wganHCAL.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/wganHCAL.py b/wganHCAL.py
index 288d1d0..822875f 100644
--- a/wganHCAL.py
+++ b/wganHCAL.py
@@ -251,8 +251,8 @@ def run(args):
print ("loading data")
#dataset = HDF5Dataset('/eos/user/e/eneren/scratch/40GeV40k.hdf5', transform=None, train_size=40000)
- #dataset = HDF5Dataset('/eos/user/e/eneren/scratch/60GeV20k.hdf5', transform=None, train_size=20000)
- dataset = HDF5Dataset('/eos/user/e/eneren/scratch/4060GeV.hdf5', transform=None, train_size=60000)
+ dataset = HDF5Dataset('/eos/user/e/eneren/scratch/50GeV75k.hdf5', transform=None, train_size=75000)
+ #dataset = HDF5Dataset('/eos/user/e/eneren/scratch/4060GeV.hdf5', transform=None, train_size=60000)
if args.world_size > 1:
--
GitLab