diff options
| author | Taesung Park <taesung_park@berkeley.edu> | 2017-04-20 02:43:45 -0700 |
|---|---|---|
| committer | Taesung Park <taesung_park@berkeley.edu> | 2017-04-20 02:43:45 -0700 |
| commit | 07274cd211e6fe180b8287eb65519e66f87d3e2a (patch) | |
| tree | 785e9cbe8cbed5cd9daf7f30be8c13f912723b5c /train.py | |
| parent | 685cfc40f5e9bb529babc2e5ebc1026b864fd119 (diff) | |
| parent | c41ab7acdb1e9191d91991d6461ddc4dff3243f7 (diff) | |
Merge branch 'master' of https://github.com/ruotianluo/pytorch-CycleGAN-and-pix2pix into ruotianluo-master
Diffstat (limited to 'train.py')
| -rw-r--r-- | train.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -32,7 +32,7 @@ for epoch in range(1, opt.niter + opt.niter_decay + 1): errors = model.get_current_errors() visualizer.print_current_errors(epoch, epoch_iter, errors, iter_start_time) if opt.display_id > 0: - visualizer.plot_current_errors(epoch, epoch_iter, opt, errors) + visualizer.plot_current_errors(epoch, float(epoch_iter)/num_train, opt, errors) if total_steps % opt.save_latest_freq == 0: print('saving the latest model (epoch %d, total_steps %d)' % |