八谷大岳の覚え書きブログ
id:hirotaka_hachiya
tensorflow2.2のsubclassingによるautoencoderの実装
以下を参考に、tensorflow2.2以降で許可されたtrain_step, test_stepおよびpredict_stepのオーバーライドを用いて、Subclassing APIでautoencoderを実装してみた。 hirotaka-hachiya.hatenablog.com以下は、全体のコードである。元画像と復元画像の平均二乗誤差を最小化するように学習している。 今回から新たに、tf.keras…