小编Tho*_*tin的帖子

无法将 TensorFlow Keras LSTM 模型保存为 SavedModel 格式

系统信息:

操作系统平台和发行版(例如 Linux Ubuntu 16.04):Windows 10

TensorFlow 从(源代码或二进制文件)安装:pip 安装

TensorFlow 版本(使用下面的命令):v2.0.0-rc2-26-g64c3d382ca 2.0.0

Python版本:3.7.1

错误:

无法将 TensorFlow Keras LSTM 模型保存为 SavedModel 格式以导出到 Google Cloud 存储桶。

错误信息:

ValueError:试图保存一个函数 b'__inference_lstm_2_layer_call_fn_36083',它引用了一个不是简单常量的符号 Tensor Tensor("dropout/mul_1:0", shape=(None, 1280), dtype=float32)。这不受支持。

代码:

import tensorflow as tf
import os
import cv2
import numpy as np
import matplotlib.pyplot as plt
import tqdm
import datetime
from sklearn.preprocessing import LabelBinarizer 

model = tf.keras.Sequential([
    tf.keras.layers.Masking(mask_value=0.),
    tf.keras.layers.LSTM(512, dropout=0.5, recurrent_dropout=0.5),
    tf.keras.layers.Dense(256, activation='relu'),
    tf.keras.layers.Dropout(0.5),
    tf.keras.layers.Dense(len(LABELS), activation='softmax')
])

model.compile(loss='categorical_crossentropy',
              optimizer='rmsprop',
              metrics=['accuracy', 'top_k_categorical_accuracy'])

test_file = 'C:/.../testlist01.txt'
train_file = 'C:/.../trainlist01.txt' …
Run Code Online (Sandbox Code Playgroud)

python lstm keras tensorflow2.0

6
推荐指数
1
解决办法
1269
查看次数

标签 统计

keras ×1

lstm ×1

python ×1

tensorflow2.0 ×1