1. เรียนรู้
    2. /
  2. Courses
    3. /
  3. PyTorchで学ぶIntroduction to Deep Learning
Connected

Exercises

Dropout を試してみる

Dropout は、学習中に一部の出力値をランダムにゼロにすることで、過学習を防ぐのに役立ちます。この演習では、Dropout を含むシンプルなニューラルネットワークを構築し、学習モードと評価モードでの挙動の違いを確認します。

torch.nn パッケージは nn として読み込まれており、features はすでに用意されています。

คำแนะนำ 1 / 2

undefined XP
    1
    2
  • 要素をゼロにする確率を 50% に設定した dropout レイヤー をモデルに追加します。
  • モデルを 学習モード に設定し、features を使って予測を生成し、出力を output_train に保存します。