[英]tensorflow.python.framework.errors_impl.InvalidArgumentError in python code
I was trying to add tensorboard to a sentdex tutorial , and got the error tensorflow.python.framework.errors_impl.InvalidArgumentError. 我试图将tensorboard添加到sentdex教程 ,并得到错误tensorflow.python.framework.errors_impl.InvalidArgumentError。 I'm rather new to tensorflow so I don't know what's wrong.
我对张力流很新,所以我不知道出了什么问题。
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')
问题是time.time代码没有()。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.