From 4c36a517316c23b585594795cb7fe44464a2436b Mon Sep 17 00:00:00 2001 From: StevenLiuWen Date: Tue, 13 Mar 2018 06:43:48 -0400 Subject: update --- Codes/constant.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'Codes/constant.py') diff --git a/Codes/constant.py b/Codes/constant.py index f6f78df..b4fc7e3 100644 --- a/Codes/constant.py +++ b/Codes/constant.py @@ -92,7 +92,7 @@ const.EVALUATE = args.evaluate # network constants const.HEIGHT = 256 const.WIDTH = 256 -const.FLOWNET_CHECKPOINT = 'models/pretrains/flownet-SD.ckpt-0' +const.FLOWNET_CHECKPOINT = 'checkpoints/pretrains/flownet-SD.ckpt-0' const.FLOW_HEIGHT = 384 const.FLOW_WIDTH = 512 @@ -136,9 +136,9 @@ if args.snapshot_dir: const.SNAPSHOT_DIR = args.snapshot_dir print(const.SNAPSHOT_DIR) else: - const.SNAPSHOT_DIR = get_dir(os.path.join('models', const.SAVE_DIR + '_' + args.snapshot_dir)) + const.SNAPSHOT_DIR = get_dir(os.path.join('checkpoints', const.SAVE_DIR + '_' + args.snapshot_dir)) else: - const.SNAPSHOT_DIR = get_dir(os.path.join('models', const.SAVE_DIR)) + const.SNAPSHOT_DIR = get_dir(os.path.join('checkpoints', const.SAVE_DIR)) if args.summary_dir: const.SUMMARY_DIR = get_dir(os.path.join('summary', const.SAVE_DIR + '_' + args.summary_dir)) -- cgit v1.2.3-70-g09d2