1. Обучение
  2. /
  3. Курса
  4. /
  5. PyTorch で学ぶ画像向け Deep Learning

Connected

упражнение

新しい畳み込み層を追加する

プロジェクトリードから新しい CNN モデルが渡されました。まずはモデルのアーキテクチャを確認し、新しい畳み込み層を追加してみましょう。

モデルは CNNModel として利用可能です。torch と、torch.nn は nn としてすでにインポートされています。

Инструкции

100 XP
  • CNNModel クラスからモデルをインスタンス化し、畳み込み層にアクセスしてください。
  • 既存層の out_channels を新しい層の in_channels に設定し、out_channels を 32、stride と padding をともに 1、kernel_size を 3 にした新しい畳み込み層を作成し、conv2 に代入してください。
  • 新しい層をモデルに追加し、名前は "conv2" としてください。