search

MokkeMeguru / TFGENZOO

Library about construction helper for Generative models e.g. Flow-based Model with Tensorflow 2.x.
https://mokkemeguru.github.io/TFGENZOO/
12 stars 2 forks source link

Conditional Scaling Layer の作成 #78

Closed MokkeMeguru closed 4 years ago

MokkeMeguru commented 4 years ago

Conditional AffineCoupling へ突っ込む conditional を再スケーリングするための手法

https://github.com/VLL-HD/conditional_invertible_neural_networks/blob/45dc7250ebfecf10d1a278edafde0fe899f30aa1/colorization_cINN/model.py#L32-L34

正直、 h のサイズで autocast させてしまっても良いかもしれない…

MokkeMeguru commented 4 years ago

というかこれ、全ての flow module ごとに独立ってマ？ 後半ほど level 深くてきっつい

MokkeMeguru commented 4 years ago

そもそもFlow-based Model側で処理しなくても良さそう