python代码中的tensorflow.python.framework.errors_impl.InvalidArgumentError

时间:2019-03-24 15:32:18

标签: python-3.x tensorflow tf.keras

我试图将张量板添加到sentdex tutorial中,并出现错误tensorflow.python.framework.errors_impl.InvalidArgumentError。我是tensorflow的新手,所以我不知道怎么了。

import tensorflow as tf
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense, Dropout, LSTM
from tensorflow.keras.callbacks import TensorBoard
import time

NAME = f"MNIST-{time.time}"

mnist = tf.keras.datasets.mnist

tensorboard = TensorBoard(log_dir='logs/{}'.format(NAME))

(x_train, y_train), (x_test, y_test) = mnist.load_data()

x_train = x_train/255
x_test = x_test/255

model = Sequential()

model.add(LSTM(128, input_shape=(x_train.shape[1:]), activation='relu', return_sequences=True))
model.add(Dropout(0.2))

model.add(LSTM(128, activation='relu'))
model.add(Dropout(0.2))

model.add(Dense(32, activation='relu'))
model.add(Dropout(0.2))

model.add(Dense(10, activation='softmax'))

opt = tf.keras.optimizers.Adam(lr=1e-3, decay=1e-5)

model.compile(loss='sparse_categorical_crossentropy', optimizer=opt, metrics=['accuracy'])

tensorboard = TensorBoard(log_dir=f'logs/{NAME}')
model.fit(x_train, y_train, epochs=2, validation_data=(x_test, y_test), callbacks = [tensorboard])

model.save('MNIST')

1 个答案:

答案 0 :(得分:0)

问题是time.time代码没有()。