Модель Python 3.5 Tensorflow ValueError

Question

import tensorflow as tf
import numpy as np
from keras.preprocessing import image
import cv2
import matplotlib.pyplot as plt

mnist = tf.keras.datasets.fashion_mnist
(training_images, training_labels), (test_images, test_labels) = mnist.load_data()

training_images = training_images.reshape(60000, 28, 28, 1)
training_images = training_images / 255.0
test_images = test_images.reshape(10000, 28, 28, 1)
test_images = test_images / 255.0

model = tf.keras.Sequential([
    tf.keras.layers.Flatten(),
    tf.keras.layers.Dense(128, activation="relu"),
    tf.keras.layers.Dense(10, activation="softmax")
])

model.compile(optimizer="adam", loss="sparse_categorical_crossentropy", metrics=["accuracy"])
model.fit(training_images, training_labels, epochs=20)

Я следую учебному пособию по искусственному интеллекту на YouTube. Оттуда я получил этот код. В учебном пособии они использовали Google CoLab, и все работало нормально. Поэтому, когда я запускаю код, последняя строка выдает ошибку: ValueError: Attempt to convert a value (9) with an unsupported type (<class 'numpy.uint8'>) to a Tensor. Кто-нибудь знает, что я пропустил или как я могу исправить ошибку?