Python3 学習10回につき1回学習結果と重みを保存したい

Question

前提・実現したいこと

VAEの実装をしており、画像5万枚の学習を100回(=500枚)に分けて行うコードを実装しています。
学習モデルと重みについては、学習10回につき1回保存するようにしたいと思い以下の通りコードを書いたのですが、
100回の学習が終わったあとに確認したところ、学習モデルと重みは1回しか保存がされていませんでした。
恐らくif文がうまく機能しておらず、上書き保存をしてしまい保存データが1つしかないのだと思うのですが、この場合コードをどのように書き換えたらいいか、ご存じの方がいらっしゃいましたらご教示いただけないでしょうか。

該当のソースコード

train_model_name = 'vae_cnn_model'
train_weights_name = 'vae_cnn_weights'

for i in range(100):
    x_train = np.load('./'+'image'+'{0:04d}'.format(i)+'.npy') # iの増加に応じて、新しいnpyファイルを読み込む
    x_train = np.reshape(x_train, [-1, original_dim])
    x_test = np.reshape(x_test, [-1, original_dim])
    x_train = x_train.astype('float32') / 255
    x_test = x_test.astype('float32') / 255
    print(x_train.shape)#形状の確認　
    
    
    # train the autoencoder
    history = vae.fit(x_train,
                epochs=epochs,
                batch_size=batch_size,
                validation_data=(x_test, None))
    if i % 10 == 0: #10回に１回学習モデルと重みを保存する
          save_model_name = os.path.join('.', train_model_name + '{0:03d}'.format(i)+'.json')
          save_weights_name = os.path.join('.', train_weights_name +'{0:03d}'.format(i)+'.h5')
          open(save_model_name, "w").write(vae.to_json())#open('vae_cnn.json', "w").write(vae.to_json()) #モデルの保存
          vae.save_weights(save_weights_name) #vae.save_weights('vae_cnn.h5') #学習結果の保存
          
          plot_results(models, data, batch_size=batch_size, model_name="vae_cnn")
          plot_history(history)   

試したこと

こちらのサイトで同様の質問がないか確認しましたが、見つかりませんでした。

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

linux PCからanacondaの仮想環境を使ってjupyter notebookで実装しています。
コードが長いため、該当の部分のみ記載しておりますが、不足している情報がありましたらお手数ですがお知らせいただけますと幸いです。

Answer 1

open(save_model_name, "w").write(vae.to_json())

となっていてcloseメソッドが呼び出されていないのが気になります。

open(save_model_name, "w").write(vae.to_json()).close()

としてみてください。

余談ですがwithを使うとファイルの閉じ忘れ防止になります。

with open(save_model_name, "w") as f:
    f.write(vae.to_json())

Comments

[SuzuAya]

ご丁寧にご回答頂きありがとうございました。
closeし忘れていたんですね。。今後はwithを使うように致します。大変参考になりました。