Closed MokkeMeguru closed 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 させてしまっても良いかもしれない…
というかこれ、全ての flow module ごとに独立ってマ? 後半ほど level 深くてきっつい
そもそもFlow-based Model側で処理しなくても良さそう
Conditional AffineCoupling へ突っ込む conditional を再スケーリングするための手法
https://github.com/VLL-HD/conditional_invertible_neural_networks/blob/45dc7250ebfecf10d1a278edafde0fe899f30aa1/colorization_cINN/model.py#L32-L34
正直、 h のサイズで autocast させてしまっても良いかもしれない…