windows 下 keras.utils.plot_model报错问题

1、下载graphviz.msi安装包,添加Bin目录到Path系统变量

下载地址:https://graphviz.gitlab.io/_pages/Download/Download_windows.html

2、如果出现'dict' object has no attribute 'name' 错误:

https://github.com/tensorflow/tensorflow/issues/38988

添加 model._layers = [layer for layer in model._layers if not isinstance(layer, dict)] 这一行

再执行keras.utils.plot_model(model, "my_first_model_with_shape_info.png", show_shapes=True)

 

注意生成图片要放在模型编译之前

你可能感兴趣的:(windows 下 keras.utils.plot_model报错问题)