From 2f2b73b306e2dd7fe4f72f64f9ba479abc48a614 Mon Sep 17 00:00:00 2001
From: Engin Eren <engin.eren@desy.de>
Date: Mon, 25 Apr 2022 14:16:46 +0200
Subject: [PATCH] 40 GeV, drop last true
---
wgan.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/wgan.py b/wgan.py
index 29e9498..d3e7a36 100644
--- a/wgan.py
+++ b/wgan.py
@@ -267,7 +267,7 @@ def run(args):
sampler = DistributedSampler(dataset, shuffle=True)
- train_loader = DataLoader(dataset, batch_size=args.batch_size, sampler=sampler, num_workers=1, pin_memory=False)
+ train_loader = DataLoader(dataset, batch_size=args.batch_size, sampler=sampler, num_workers=1, drop_last=True, pin_memory=False)
--
GitLab