From cc4df23a0526d02b58e60eedf340480ba79dfb90 Mon Sep 17 00:00:00 2001
From: Engin Eren <engin.eren@desy.de>
Date: Wed, 27 Apr 2022 16:56:35 +0200
Subject: [PATCH] 40 and 60 GeV
---
wgan.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/wgan.py b/wgan.py
index 22c69b3..502f13f 100644
--- a/wgan.py
+++ b/wgan.py
@@ -263,7 +263,7 @@ 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/50GeV75k.hdf5', transform=None, train_size=75000)
+ dataset = HDF5Dataset('/eos/user/e/eneren/scratch/4060GeV.hdf5', transform=None, train_size=60000)
sampler = DistributedSampler(dataset, shuffle=True)
--
GitLab