Я пытался добавить тензорную доску к учебнику sentdex и получил ошибку tenorflow.python.framework.errors_impl.InvalidArgumentError.Я довольно новичок в тензорном потоке, поэтому я не знаю, в чем дело.
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')