Ошибка при добавлении слоя keras

Начал изучать keras и написал простой код:

from keras.datasets import mnist
from keras import models, layers
from keras.utils import to_categorical

(train_imges, train_lables), (test_imges, test_lables) = mnist.load_data()

network = models.Sequential()
network.add(layers.Dense(512, activation='relu', input_shape=(28 * 28,)))
network.add(layers.Dense(10, activation="softmax"))

network.compile(optimizer="rmsprop", loss='categorical_crossent', metric=["accuracy"])

train_imges = train_imges.reshape((60000, 28 * 28)) 
train_imges = train_imges.astype('float') / 255 

train_lables = to_categorical(train_lables)
test_lables = to_categorical(test_lables)

network.fit(train_imges, train_lables, epochs=5, batch_size=128)

Но он выдает ошибку, в инете ничего вменяемого по этому не нашел.

Traceback (most recent call last):
  File "D:/Inf/python/deep_learning_start.py", line 10, in <module>
    network.add(layers.Dense(512, activation='relu', input_shape=(28 * 28,)))
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\training\tracking\base.py", line 456, in _method_wrapper
    result = method(self, *args, **kwargs)
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\keras\engine\sequential.py", line 198, in add
    layer(x)
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\keras\engine\base_layer.py", line 897, in __call__
    self._maybe_build(inputs)
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\keras\engine\base_layer.py", line 2416, in _maybe_build
    self.build(input_shapes)  # pylint:disable=not-callable
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\keras\layers\core.py", line 1158, in build
    self.kernel = self.add_weight(
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\keras\engine\base_layer.py", line 560, in add_weight
    variable = self._add_variable_with_custom_getter(
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\training\tracking\base.py", line 738, in _add_variable_with_custom_getter
    new_variable = getter(
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\keras\engine\base_layer_utils.py", line 129, in make_variable
    return tf_variables.VariableV1(
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\ops\variables.py", line 259, in __call__
    return cls._variable_v1_call(*args, **kwargs)
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\ops\variables.py", line 205, in _variable_v1_call
    return previous_getter(
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\ops\variables.py", line 198, in <lambda>
    previous_getter = lambda **kwargs: default_variable_creator(None, **kwargs)
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\ops\variable_scope.py", line 2584, in default_variable_creator
    return resource_variable_ops.ResourceVariable(
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\ops\variables.py", line 263, in __call__
    return super(VariableMetaclass, cls).__call__(*args, **kwargs)
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\ops\resource_variable_ops.py", line 1423, in __init__
    self._init_from_args(
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\ops\resource_variable_ops.py", line 1577, in _init_from_args
    handle = eager_safe_variable_handle(
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\ops\resource_variable_ops.py", line 242, in eager_safe_variable_handle
    return _variable_handle_from_shape_and_dtype(
  File "C:\Users\Dolgo\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\ops\resource_variable_ops.py", line 179, in _variable_handle_from_shape_and_dtype
    handle_data.shape_and_type.append(
AttributeError: 'RepeatedCompositeFieldContainer' object has no attribute 'append'