From 8635791f0699a98e70de7b3b096ab60c9cdce443 Mon Sep 17 00:00:00 2001
From: whs <wanghaoshuang@baidu.com>
Date: Mon, 15 Mar 2021 17:35:29 +0800
Subject: [PATCH] Fix pruning export (#2350)

---
 slim/prune/export_model.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/slim/prune/export_model.py b/slim/prune/export_model.py
index dd3c0ae46..d8427d79a 100644
--- a/slim/prune/export_model.py
+++ b/slim/prune/export_model.py
@@ -63,6 +63,9 @@ def main():
             test_fetches = model.test(feed_vars)
     infer_prog = infer_prog.clone(True)
 
+    exe.run(startup_prog)
+    checkpoint.load_checkpoint(exe, infer_prog, cfg.weights)
+
     pruned_params = FLAGS.pruned_params
     assert (
         FLAGS.pruned_params is not None
@@ -90,13 +93,9 @@ def main():
     logger.info("pruned FLOPS: {}".format(
         float(base_flops - pruned_flops) / base_flops))
 
-    exe.run(startup_prog)
-    checkpoint.load_checkpoint(exe, infer_prog, cfg.weights)
-
     dump_infer_config(FLAGS, cfg)
     save_infer_model(FLAGS, exe, feed_vars, test_fetches, infer_prog)
 
-
 if __name__ == '__main__':
     enable_static_mode()
     parser = ArgsParser()
-- 
GitLab