From 8b4cb9ff0b5f57bfd53b673608438353db3ae22a Mon Sep 17 00:00:00 2001
From: Engin Eren <engin.eren@desy.de>
Date: Mon, 4 Apr 2022 11:07:32 +0200
Subject: [PATCH] clean up
---
wgan.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/wgan.py b/wgan.py
index 1e762aa..95cf6db 100644
--- a/wgan.py
+++ b/wgan.py
@@ -103,7 +103,7 @@ def train(args, aD, aG, device, train_loader, optimizer_d, optimizer_g, epoch):
aG.train()
#aE.eval()
- print("Generator training started")
+ #print("Generator training started")
optimizer_g.zero_grad()
@@ -130,6 +130,10 @@ def train(args, aD, aG, device, train_loader, optimizer_d, optimizer_g, epoch):
g_cost.backward()
optimizer_g.step()
+ if batch_idx % args.log_interval == 0 :
+ print('Generator --> Train Epoch: {} [{}/{} ({:.0f}%)]\tloss={:.4f}'.format(
+ epoch, batch_idx * len(data), len(train_loader.dataset),
+ 100. * batch_idx / len(train_loader), g_cost.item()))
@@ -212,8 +216,6 @@ def parse_args():
def run(args):
# Training settings
-
- print ("beginning of python script")
use_cuda = not args.no_cuda and torch.cuda.is_available()
if use_cuda:
--
GitLab