タグ

ブックマーク / blog.shikoan.com (1)

  • tensorflow.kerasでKeras方式のhdf5で重みを保存する方法

    従来のKerasで係数を保存すると「hdf5」形式で保存されたのですが、TPU環境などでTensorFlowのKerasAPIを使うと、TensorFlow形式のチェックポイントまるごと保存で互換性の面で困ったことがおきます。従来のKerasのhdf5形式で保存する方法を紹介します。 サンプルコード これはGoogle ColabのTPUでMNISTを分類するコードです。 import tensorflow as tf from keras.datasets import mnist from tensorflow.keras.layers import Dense, Input from tensorflow.keras.models import Model from tensorflow.keras.optimizers import Adam from keras.utils im

    tensorflow.kerasでKeras方式のhdf5で重みを保存する方法
    macj_jp
    macj_jp 2019/11/20
  • 1