From d101ebf83288b7dc994118ee6dd8857583beaecb Mon Sep 17 00:00:00 2001 From: tingchunw Date: Wed, 6 Dec 2017 02:40:43 +0000 Subject: fix tab issue --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'train.py') diff --git a/train.py b/train.py index 4965481..231662c 100755 --- a/train.py +++ b/train.py @@ -79,7 +79,7 @@ for epoch in range(start_epoch, opt.niter + opt.niter_decay + 1): ############## Display results and errors ########## ### print out errors if total_steps % opt.print_freq == 0: - errors = {k: v.data[0] if not isinstance(v, (int,long,float)) else v for k, v in loss_dict.items()} + 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) @@ -98,7 +98,7 @@ for epoch in range(start_epoch, opt.niter + opt.niter_decay + 1): np.savetxt(iter_path, (epoch, epoch_iter), delimiter=',', fmt='%d') # end of epoch - iter_end_time = time.time() + iter_end_time = time.time() print('End of epoch %d / %d \t Time Taken: %d sec' % (epoch, opt.niter + opt.niter_decay, time.time() - epoch_start_time)) -- cgit v1.2.3-70-g09d2