深度学习一些工程技巧

使用plot_model绘制keras模型图

安装如下依赖:
pip install pydot
pip install graphviz
conda install graphviz python-graphviz
定义完模型后,用如下代码绘制:

from keras.utils import plot_model
plot_model(self.model,to_file="model.png",show_shapes=True)

输出模型图的一部分如下图所示:
示例

定义好网络结构之后,冻结网络结构以防止修改

可使用如下代码:

sess.graph.finalize() 

没有更多推荐了,返回首页

私密
私密原因:
请选择设置私密原因
  • 广告
  • 抄袭
  • 版权
  • 政治
  • 色情
  • 无意义
  • 其他
其他原因:
120
出错啦
系统繁忙,请稍后再试

关闭