以下では、tensorflowのバージョン2.1を前提にSequential、FunctionalおよびSubclassing APIを用いた実装方法についておおざっぱにまとめた。 hirotaka-hachiya.hatenablog.com tensorflow 2.1のSubclassing APIでは、自由度が高い分学習まわりでfor文でepochを回すなどコードが煩雑になり一貫性の無いところが残念だった…

八谷大岳の覚え書きブログ

train_stepのオーバーライドによるsubclassingの簡単化