1. Learn
    2. /
  2. Courses
    3. /
  3. PyTorchによる中級ディープラーニング
Connected

Exercise

RNN の学習ループ

電力消費量を予測するモデルを学習させましょう！

これまでに定義した LSTM ネットワークはすでにインスタンス化されて net に代入されています。以前に作成した dataloader_train も用意されています。さらに、torch.nn は nn としてインポート済みです。

この演習では、学習が正しく進んでいるかを確認するために、エポック数は 3 にとどめます。では始めましょう！

Instructions

100 XP
  • 平均二乗誤差（Mean Squared Error）損失を設定し、criterion に代入します。
  • seqs を (バッチサイズ, シーケンス長, 特徴量数)、今回のケースでは (32, 96, 1) にリシェイプし、結果を seqs に再代入します。
  • モデルに seqs を渡して outputs を得ます。
  • これまでに計算した量に基づいて損失を計算し、loss に代入します。