1. 学ぶ
    2. /
  2. コース
    3. /
  3. PyTorch で学ぶ画像向け Deep Learning
Connected

演習

Sequential ブロックを作成する

畳み込みレイヤーのブロックを作成して、バイナリ分類用の CNN モデルのテンプレートを作り直すことにしました。これにより、複数のレイヤーを順番に積み重ねられます。改良したこのモデルを使えば、さまざまな CNN アーキテクチャを簡単に設計できるようになります。

torch と、torch.nn を nn としてインポート済みです。

指示

100 XP
  • __init__() メソッド内で、畳み込みレイヤーのブロックを定義し、self.conv_block に代入します。
  • forward() の処理で、定義した畳み込みブロックに入力を通します。