From c0f267c6f09fbd025e590be5b6c0ef35f2b0b991 Mon Sep 17 00:00:00 2001 From: Edgar Riba Date: Tue, 7 Mar 2017 18:44:03 +0100 Subject: [PATCH] show epoch progress correctly --- main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 948a6f3..329ff42 100644 --- a/main.py +++ b/main.py @@ -214,8 +214,8 @@ for epoch in range(opt.niter): optimizerG.step() gen_iterations += 1 - print('[%d/%d][%d/%d] Loss_D: %f Loss_G: %f Loss_D_real: %f Loss_D_fake %f' - % (epoch, opt.niter, gen_iterations, len(dataloader), + print('[%d/%d][%d/%d][%d] Loss_D: %f Loss_G: %f Loss_D_real: %f Loss_D_fake %f' + % (epoch, opt.niter, i, len(dataloader), gen_iterations, errD.data[0], errG.data[0], errD_real.data[0], errD_fake.data[0])) if gen_iterations % 500 == 0: vutils.save_image(real_cpu, '{0}/real_samples.png'.format(opt.experiment))