shine-Notes

ゆるふわ思考ダンプ

2020-09-01から1ヶ月間の記事一覧

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…