keras モデルを読み込めない

Question

前提・実現したいこと

kerasで学習したモデルを保存し，予測に使うためにモデルを読み込もうとするとエラーが出てしまいます．

発生している問題・エラーメッセージ

Invalid initialization: my_init

該当のソースコード

# 一部記載
from keras import initializations
from keras.models import Sequential
from keras.models import load_model

def my_init(shape, name=None):
    return initializations.normal(shape, scale=0.1, name=name)

# 学習
def learn():
    #前処理は省略
    model = Sequential()
    model.add(Convolution2D(32, 3, 3, init=my_init, input_shape=input_shape))
    ##
    #中略
    ##
    model.compile(loss='categorical_crossentropy', optimizer='adadelta', metrics=['accuracy'])

    hist = model.fit_generator(datagen.flow(X_train, Y_train, batch_size=16), samples_per_epoch=X_train.shape[0],
                        nb_epoch=10, validation_data=(X_test, Y_test))

    model.save("model.h5")

def test():
    model = load_model("model.h5")

    

セルで実行

>>> learn()
>>> test() # ここで上記のエラーが出ます

補足情報（FW/ツールのバージョンなど）

こちらのサイトのコードを参考にしています．
https://qiita.com/neet-AI/items/2b3d7f743e4d6c6d8e10

実行はGoogle Colabで行っていますが，バージョンは上記のコードに合わせて以下のように．変更しています．
python:3.6.9
keras:1.1.0
theano:0.8.2
scikit-learn:0.18.0
scipy:1.2.0

古いバージョンで申し訳ございませんが，よろしくお願いいたします．

Answer 1

kerasを2.0.0にバージョンアップし，

model.add(Convolution2D(32, 3, 3, init=my_init, input_shape=input_shape))

を

model.add(Conv2D(32, (3, 3), kernel_initializer=initializers.VarianceScaling(scale=0.1), input_shape=input_shape))

に変更したらできました．