Skip to content

B
book

PaddlePaddle / book

通知 17
Star 4
Fork 0
B
book
提交 b25a435b 编写于 作者: F fengjiayi
浏览文件
操作

Format

上级 82fdd3f9
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
03.image_classification/train.py
浏览文件 @ b25a435b
...@@ -79,7 +79,7 @@ def main(): ...@@ -79,7 +79,7 @@ def main():
# Create trainer # Create trainer
trainer = paddle.trainer.SGD( trainer = paddle.trainer.SGD(
cost=cost, parameters=parameters, update_equation=momentum_optimizer) cost=cost, parameters=parameters, update_equation=momentum_optimizer)
# Save the inference topology to protobuf. # Save the inference topology to protobuf.
inference_topology = paddle.topology.Topology(layers=out) inference_topology = paddle.topology.Topology(layers=out)
with open("inference_topology.pkl", 'wb') as f: with open("inference_topology.pkl", 'wb') as f:
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或