From a87a78e35445c20cf89052c4c591dd4e878ec19a Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 24 Sep 2018 13:10:26 +0200 Subject: oops log --- train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'train.py') diff --git a/train.py b/train.py index d0481e4..733282a 100755 --- a/train.py +++ b/train.py @@ -37,7 +37,7 @@ dataset_size = len(data_loader) print('dataset size = %d' % dataset_size) model = create_model(opt) -# visualizer = Visualizer(opt) +visualizer = Visualizer(opt) total_steps = (start_epoch-1) * dataset_size + epoch_iter @@ -90,8 +90,8 @@ for epoch in range(start_epoch, start_epoch + opt.niter): if total_steps % opt.print_freq == print_delta: errors = {k: v.data[0] if not isinstance(v, int) else v for k, v in loss_dict.items()} t = (time.time() - iter_start_time) / opt.batchSize - #visualizer.print_current_errors(epoch, epoch_iter, errors, t) - #visualizer.plot_current_errors(errors, total_steps) + visualizer.print_current_errors(epoch, epoch_iter, errors, t) + visualizer.plot_current_errors(errors, total_steps) ### display output images #if save_fake: -- cgit v1.2.3-70-g09d2