From 30636a2928551c00fe5adf823e03c7315425797c Mon Sep 17 00:00:00 2001
From: Liufang Sang <slf12thuss@163.com>
Date: Wed, 25 Mar 2020 00:52:34 -0500
Subject: [PATCH] check path exists (#190)

---
 demo/quant/quant_aware/train.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/demo/quant/quant_aware/train.py b/demo/quant/quant_aware/train.py
index 996b0405..81016e29 100644
--- a/demo/quant/quant_aware/train.py
+++ b/demo/quant/quant_aware/train.py
@@ -148,6 +148,9 @@ def compress(args):
     exe = fluid.Executor(place)
     exe.run(fluid.default_startup_program())
 
+    assert os.path.exists(
+        args.pretrained_model), "pretrained_model doesn't exist"
+
     if args.pretrained_model:
 
         def if_exist(var):
@@ -241,11 +244,11 @@ def compress(args):
                 exe,
                 dirname=os.path.join(args.checkpoint_dir, 'best_model'),
                 main_program=val_program)
-
-    fluid.io.load_persistables(
-        exe,
-        dirname=os.path.join(args.checkpoint_dir, 'best_model'),
-        main_program=val_program)
+    if os.path.exists(os.path.join(args.checkpoint_dir, 'best_model')):
+        fluid.io.load_persistables(
+            exe,
+            dirname=os.path.join(args.checkpoint_dir, 'best_model'),
+            main_program=val_program)
     ############################################################################################################
     # 3. Freeze the graph after training by adjusting the quantize
     #    operators' order for the inference.
-- 
GitLab