diff options
Diffstat (limited to 'Codes/constant.py')
| -rw-r--r-- | Codes/constant.py | 6 |
1 files changed, 3 insertions, 3 deletions
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)) |