diff options
| -rwxr-xr-x | train.py | 6 |
1 files changed, 3 insertions, 3 deletions
@@ -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: |