From 908e0644bacad5850798c52c1c050f5f733040a2 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Fri, 7 Sep 2018 15:24:23 +0200 Subject: augment plus 2 --- augment.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'augment.py') diff --git a/augment.py b/augment.py index b01f04c..4aab5f1 100755 --- a/augment.py +++ b/augment.py @@ -67,7 +67,7 @@ if data_opt.tag == '': else: tag = data_opt.tag -if opt.current_epoch == 'latest': +if opt.which_epoch == 'latest': iter_path = os.path.join(opt.checkpoints_dir, opt.name, 'iter.txt') if os.path.exists(iter_path): try: @@ -78,7 +78,7 @@ if opt.current_epoch == 'latest': else: current_epoch, epoch_iter = 1, 0 else: - current_epoch = opt.current_epoch + current_epoch = opt.which_epoch epoch_id = "{:02d}_{:04d}_{:04d}".format(current_epoch, data_opt.augment_take, data_opt.augment_make) opt.render_dir = os.path.join(opt.results_dir, opt.name, epoch_id) -- cgit v1.2.3-70-g09d2