From b122ac967bf53f62a3fc59176b5f75cfe9f69618 Mon Sep 17 00:00:00 2001
From: Engin Eren <engin.eren@desy.de>
Date: Sun, 24 Apr 2022 00:33:27 +0200
Subject: [PATCH] 60GeV
---
wgan.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/wgan.py b/wgan.py
index b3676f2..7322b66 100644
--- a/wgan.py
+++ b/wgan.py
@@ -262,7 +262,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/50GeV75k.hdf5', transform=None, train_size=75000)
+ dataset = HDF5Dataset('/eos/user/e/eneren/scratch/60GeV20k.hdf5', transform=None, train_size=20000)
sampler = DistributedSampler(dataset, shuffle=True)
train_loader = DataLoader(dataset, batch_size=args.batch_size, sampler=sampler, num_workers=1, pin_memory=False)
--
GitLab