提交 44cf727d 编写于 作者: Q qiaolongfei

save model proto as file

上级 c97b2ddd
......@@ -21,6 +21,10 @@ def main():
trainer = paddle.trainer.SGD(
cost=cost, parameters=parameters, update_equation=optimizer)
# save model proto as file
with open("model.proto", "w") as f:
f.write(str(trainer.__topology_in_proto__))
feeding = {'x': 0, 'y': 1}
# event_handler to print training and testing info
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册