未验证 提交 6defc5c8 编写于 作者: L LielinJiang 提交者: GitHub

Merge pull request #184 from wanghaoshuang/fix_slim

Fix pruning demo
...@@ -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.
先完成此消息的编辑!
想要评论请 注册