Open MokkeMeguru opened 4 years ago
おそらく TF2.1.0から、レイヤー内で生成するオブジェクトがスタックするので、オブジェクトではなく、メソッドとして実装しないといけない。
≫訓練時間とメモリリークがヤバイ
tensorflow 2.2.0 でちゃんとうごいているような雰囲気を感じないわけじゃない…
但しTensorflow Probability のバージョン制約が厳しすぎるので依存関係をなくしたい
WARN: Tensorflow Probability 0.10.0 はインストールしてはいけない
https://github.com/openai/glow/blob/eaff2177693a5d84a1cf8ae19e8e0441715b82f8/tfops.py#L448-L450