提交 9fe84c65 编写于 作者: W wanghaoshuang

Fix pruning demo

上级 bbd53e5c
...@@ -43,7 +43,7 @@ from models.model_builder import ModelPhase ...@@ -43,7 +43,7 @@ from models.model_builder import ModelPhase
from reader import SegDataset from reader import SegDataset
from metrics import ConfusionMatrix from metrics import ConfusionMatrix
from paddleslim.prune.io import * from paddleslim.prune import load_model
def parse_args(): def parse_args():
parser = argparse.ArgumentParser(description='PaddleSeg model evalution') parser = argparse.ArgumentParser(description='PaddleSeg model evalution')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册