トレーニングループ

ついに、これまで定義してきたモデル構造と損失関数が活躍するときです。いよいよ学習を実行します。あなたの役割は、GAN のトレーニングループを実装して実行することです。注意: 実行時間を長くしないため、最初のバッチの後で break 文が入っています。

2 つのオプティマイザ disc_opt と gen_opt は、Adam() オプティマイザとして初期化済みです。先ほど定義した損失を計算する関数 gen_loss() と disc_loss() が利用できます。dataloader も用意されています。

次を思い出してください。