Skip to content

P
PaddleX

PaddlePaddle / PaddleX

通知 138
Star 4
Fork 0
P
PaddleX
提交 8d243913 编写于 作者: J jiangjiajun
浏览文件
操作

add AlexNet

上级 4509ef1d
隐藏空白更改
内联 并排
Showing with 4 addition and 3 deletion
paddlex/cv/models/load_model.py
浏览文件 @ 8d243913
...@@ -46,9 +46,10 @@ def load_model(model_dir, fixed_input_shape=None): ...@@ -46,9 +46,10 @@ def load_model(model_dir, fixed_input_shape=None):
if '_Attributes' in info: if '_Attributes' in info:
if 'fixed_input_shape' in info['_Attributes']: if 'fixed_input_shape' in info['_Attributes']:
fixed_input_shape = info['_Attributes']['fixed_input_shape'] fixed_input_shape = info['_Attributes']['fixed_input_shape']
logging.info("Model already has fixed_input_shape with {}".format( if fixed_input_shape is not None:
fixed_input_shape)) logging.info("Model already has fixed_input_shape with {}".
model.fixed_input_shape = fixed_input_shape format(fixed_input_shape))
model.fixed_input_shape = fixed_input_shape
if status == "Normal" or \ if status == "Normal" or \
status == "Prune" or status == "fluid.save": status == "Prune" or status == "fluid.save":
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或