googlecolaboratoryを使ったkerasでmodel.saveができない

画像認識プログラミングレシピという本で画像認識を学習中です。
kerasでモデルを保存しようとするとエラーが出て保存できません。
以下のサイトが参考になるかなと思ったのですが、読んでもよく理解できませんでした。
https://qiita.com/taiga518/items/b2154b661e7baf56031e

文字数の関係上モデルの一部を抜粋させていただきました。
必要な情報があればコメントお願いします。

python
1model = Sequential()
2
3#学習方法を決める
4
5#(学習スタイル, 損失関数, 正解かどうかの評価方法)
6model.compile(optimizer=keras.optimizers.Adadelta(),
7             loss=keras.losses.categorical_crossentropy,
8             metrics=["accuracy"])
9
10print("反復学習回数:", EPOCHS)
11fit_record = model.fit(train_data, train_teacher_labels, batch_size=BATCH_SIZE, epochs=EPOCHS, verbose=1, validation_data=(test_data, test_teacher_labels))
12
13#ニューラルネットワークの構築
14
15#畳み込み層(入力のニューロン数, 畳み込みエリアの幅と高さ, 活性化関数, 入力データの形式)
16model.add(Conv2D(32, kernel_size=(3, 3), activation="relu", input_shape=input_shape))
17model.add(Conv2D(64, (3,3), activation="relu"))
18#プーリング層
19model.add(MaxPooling2D(pool_size=(2, 2)))
20#ドロップアウト層(過学習を防ぐ)
21model.add(Dropout(0.25))
22#入力を平滑化する
23model.add(Flatten())
24#全結合層(ノード数が128個, 活性化関数)
25model.add(Dense(128, activation="relu"))
26model.add(Dropout(0.5))
27#出力層(ノーど数がnum_classes個になる)
28model.add(Dense(NUM_CLASSES, activation="softmax"))

エラーが出たところ

python
1model.save("keras-mnist-model.h5")

エラー内容

python
1---------------------------------------------------------------------------
2NotImplementedError                       Traceback (most recent call last)
3<ipython-input-90-a2ac93d46f7c> in <module>()
4----> 1 model.save("keras-mnist-model.h5")
5
68 frames
7/usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/engine/base_layer.py in get_config(self)
8    719       raise NotImplementedError('Layer %s has arguments in `__init__` and '
9    720                                 'therefore must override `get_config`.' %
10--> 721                                 self.__class__.__name__)
11    722     return config
12    723 
13
14NotImplementedError: Layer ModuleWrapper has arguments in `__init__` and therefore must override `get_config`.