From 9fe84c65e9f3474844336906267eb840bd5b1147 Mon Sep 17 00:00:00 2001
From: wanghaoshuang <wanghaoshuang@baidu.com>
Date: Wed, 11 Mar 2020 20:29:57 +0800
Subject: [PATCH] Fix pruning demo

---
 slim/prune/eval_prune.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/slim/prune/eval_prune.py b/slim/prune/eval_prune.py
index 940adce0..b8275d03 100644
--- a/slim/prune/eval_prune.py
+++ b/slim/prune/eval_prune.py
@@ -43,7 +43,7 @@ from models.model_builder import ModelPhase
 from reader import SegDataset
 from metrics import ConfusionMatrix
 
-from paddleslim.prune.io import *
+from paddleslim.prune import load_model
 
 def parse_args():
     parser = argparse.ArgumentParser(description='PaddleSeg model evalution')
-- 
GitLab