Torchの中の生成対立ネットワークモジュールには何がありますか?
Torchで使われるGAN(Generative Adversarial Networks)モジュールには、以下のような一般的なライブラリがあります。
- nn.Sequentialは、シーケンシャルネットワークを定義するための便利な方法です。
- nn.Linearは、線形関数を提供するPyTorchのクラスです。
- 畳み込み層
- nn.BatchNorm2d → ニューラルネットワーク用の2次元バッチ正規化
- nn.ReLUを日本語で言うと、nn.ランプ関数となります。
- nn.LeakyReLU を使ってください。
- シグモイド関数(nn.Sigmoid)
- nn.BCELossは、二値分類の交差エントロピー損失を計算する関数です。
- nn.L1Lossは、絶対値誤差損失関数を計算するためのPyTorchクラスです。
- nn.MSELossは、平均二乗誤差損失関数を表します。
これらのモジュールは組み合わせて、生成対抗ネットワークを構築することができ、生成器と識別器をトレーニングして、生成器がリアルなデータに近いサンプルを生成し、識別器を騙すことができます。