1. 학습
    2. /
  2. 강의
    3. /
  3. PyTorchによる中級ディープラーニング
Connected

연습 문제

バッチ正規化

モデルアーキテクチャの最終的な改善として、2つの全結合（linear）層それぞれの後にバッチ正規化層を追加してみましょう。バッチ正規化のテクニックは、学習の収束を速め、vanishing/exploding gradients の問題からモデルを保護します。

torch.nn と torch.nn.init はそれぞれ nn と init としてすでにインポートされています。モデルアーキテクチャを変更したら、バッチ正規化の仕組みに関する短い質問に答える準備をしてください！

지침 1/3

undefined XP
    1
    2
    3
  • 2つの BatchNorm1d 層を追加し、それぞれを self.bn1 と self.bn2 に代入します。