shine-Notes

ゆるふわ思考ダンプ

GAN

tensorflow v2 のBatchNormalizationを使うと参考書のGAN訓練(tf1.x)が失敗した話

TL;DR tf1.xベースの所謂keras-GANベースの実装をtensorflow2.xで実行すると、学習失敗する BatchNormalizationの部分をtf.compat.v1.keras.layers.BatchNormalizationに変更すると成功する 原因はkeras-GANにおけるtrainableの切り替え。v2のBatchNormaliza…

DCGANの学習済GeneratorをFlaskアプリ上で実行する

サマリ Webフレームワークのズブの素人が、ボタンクリックでGANのGeneratorが走るようなWebアプリを作ってみた。 Flaskは確かに短いコードでプロトタイプが作れるが、明示されていない所で「何をやっているか」を結局理解しないとハマるところも多い