时间戳

首页 > 解决方案 > 意外的关键字参数“show_dtype”

tensorflow - 意外的关键字参数“show_dtype”

问题描述

我正在尝试使用以下代码使用数据类型绘制我的模型：

plot_model(model, to_file='model/model.png', show_dtype=True, show_shapes=True, show_layer_names=True)

但是，show_dtype即使它出现在 TensorFlow 文档中，我也会收到一个不可接受的参数错误：https ://www.tensorflow.org/api_docs/python/tf/keras/utils/plot_model

标签： tensorflow

解决方案

这是我第一次遇到这个问题。如果您从 Anaconda Forge 而不是 Pip 之类的其他东西下载它，这似乎是由于具有较早的版本。然而，这是一个简单的修复。

基本上，您需要进入库源文件并将其编辑为 TensorFlow 文档页面上显示的当前版本。

您将从此处复制 Python 代码的 GitHub 页面的链接：https ://github.com/tensorflow/tensorflow/blob/v2.5.0/tensorflow/python/keras/utils/vis_utils.py#L278-L348

然后，前往你的库路径并将 Python 代码粘贴到那里。例如，我的路径如下C:/ProgramData/Anaconda3/envs/ml/Lib/site-packages/tensorflow/python/keras/utils/vis_utils.py：你的应该是类似的。

推荐阅读