Traceback (most recent call last): File "convert.py", line 6, in <module> model = tf.saved_model.load(saved_model_diexport_dir=saved_model_dir, tags=["serve"]) File "D:\anaconda3\envs\tf1.15\lib\site-packages\tensorflow_core\python\util\deprecation.py", line 324, in new_func return func(*args, **kwargs) TypeError: load() missing 2 required positional arguments: 'sess' and 'export_dir'
时间: 2023-09-10 18:15:29 浏览: 215
这个错误的原因是在加载 TensorFlow 的 SavedModel 时,缺少必要的参数。正确的方式是使用 `tf.compat.v1.Session()` 创建一个会话,然后将该会话传递给 `tf.compat.v1.saved_model.load()` 函数。以下是一个示例代码:
```python
import tensorflow.compat.v1 as tf
# 创建一个 TensorFlow 1.x 的会话
tf.disable_v2_behavior()
sess = tf.Session()
# 指定 SavedModel 的路径
saved_model_dir = "path/to/saved_model"
# 加载 SavedModel
model = tf.saved_model.load(sess, ["serve"], saved_model_dir)
# 关闭会话
sess.close()
```
请注意,在 TensorFlow 2.x 中,`tf.compat.v1.Session()` 已经被弃用,应该使用 `tf.compat.v1.enable_eager_execution()` 和 `tf.saved_model.load()` 来加载 SavedModel。
相关问题
Traceback (most recent call last): File "convert.py", line 6, in <module> model = tf.saved_model.load(saved_model_diexport_dir=saved_model_dir, tags=tagsr) NameError: name 'tagsr' is not defined怎么解决
这个错误是因为 `tagsr` 这个变量没有定义导致的。你需要将代码中的 `tagsr` 修改为正确的标签名称。通常情况下,可以使用 `"serve"` 作为标签名称,例如:
```
model = tf.saved_model.load(saved_model_dir=saved_model_dir, tags=["serve"])
```
如果你之前已经将标签名称定义为变量,请确保该变量被正确命名和定义。
Traceback (most recent call last): File "convert.py", line 5, in <module> model = tf.saved_model.load(saved_model_dir) File "D:\anaconda3\envs\tf1.15\lib\site-packages\tensorflow_core\python\util\deprecation.py", line 324, in new_func return func(*args, **kwargs) TypeError: load() missing 2 required positional arguments: 'tags' and 'export_dir'怎么解决
这个错误提示是因为 `tf.saved_model.load()` 方法需要传入两个必需的参数:'tags' 和 'export_dir',但是在你的代码中没有提供这些参数。建议你检查一下你的代码,确保你已经正确的传递了这两个参数。下面是一个示例代码,展示了如何正确使用 `tf.saved_model.load()` 方法:
```python
import tensorflow as tf
saved_model_dir = '/path/to/your/saved_model'
tags = [tf.saved_model.tag_constants.SERVING]
model = tf.saved_model.load(export_dir=saved_model_dir, tags=tags)
```
在这个示例代码中,我们提供了 'saved_model_dir' 参数作为导出的 SavedModel 目录的路径,同时还提供了 'tags' 参数,它是一个字符串列表,用于指定要加载的模型的标签。
阅读全文