diff --git a/wgan.py b/wgan.py
index 1e762aab3ab7676f00dd8ef49a08bc0c0ed874eb..95cf6db94cd7be2a528a7f695560cc926bdd1775 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: