diff --git a/wganHCAL.py b/wganHCAL.py
index 288d1d0317ab3d57226f3264efca37dd7711ef1f..822875f4b9792e50fac1438b9d1b44c4d9bfdf79 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: