การเพิ่ม Convolutional Layer ใหม่

หัวหน้าโปรเจกต์ได้มอบโมเดล CNN ใหม่ให้ มาดูสถาปัตยกรรมของโมเดลนี้และเพิ่ม convolutional layer ใหม่เข้าไปกัน

โมเดลพร้อมใช้งานในชื่อ CNNModel และได้ import แพ็กเกจ torch และ torch.nn ในชื่อ nn ไว้แล้ว

แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร

Deep Learning สำหรับภาพด้วย PyTorch

ดูคอร์ส

คำแนะนำการฝึกหัด

สร้างอินสแตนซ์ของโมเดลจากคลาส CNNModel และเข้าถึง convolutional layers
สร้าง convolutional layer ใหม่โดยกำหนด in_channels ให้เท่ากับ out_channels ของ layer ที่มีอยู่, out_channels เป็น 32, stride และ padding เป็น 1 ทั้งคู่ และ kernel_size เป็น 3 แล้วกำหนดให้กับตัวแปร conv2
เพิ่ม layer ใหม่เข้าไปในโมเดลโดยตั้งชื่อว่า "conv2"

แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ

ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์

# Create a model
model = ____
print("Original model: ", model)

# Create a new convolutional layer
conv2 = ____

# Append the new layer to the model
model.____(____)
print("Extended model: ", model)

แก้ไขและรันโค้ด