競プロ雑記帳

windows10 + tensorflow 2.1 で学習がうまくいかない

タイトル通りです。

この前環境構築をしてからkerasを使ってGANを作っていたのですが、学習が全然進まないので loss を見てみると generator も discriminator も loss が 0 に近くなっており、状況としてはおかしいものになっていました。
そこで他の方のコードとデータセットをお借りして試してみたところそちらもうまくいかないのでいろいろ調べていると
https://github.com/tensorflow/tensorflow/issues/36072
ここや、ここの中で言及されているほかの issue を見ているとなんとなくそれっぽいので

import tensorflow as tf
tf.keras.layers.BatchNormalization._USE_V2_BEHAVIOR = False

と付け加えてみたところ借りたコードはうまく生成されました。

import tensorflow as tf
tf.compat.v1.disable_v2_behavior()

これにしてもよさそうです。まあ、tensorflow 2.1 を導入しなければいいだけなのですが...

もし、 _USE_V2_BEHAVIOR が何なのかを知っている方がいればコメントなどをいただけるとありがたいです。