From 1afd165696a17c478a55e2b58d9e06b9568b53d5 Mon Sep 17 00:00:00 2001 From: x0905y Date: Fri, 14 Mar 2025 22:03:02 +0300 Subject: [PATCH] Update submission_template04.py --- Deep Learning/submission_template04.py | 28 +++++++++++++++++++------- 1 file changed, 21 insertions(+), 7 deletions(-) diff --git a/Deep Learning/submission_template04.py b/Deep Learning/submission_template04.py index 498fa3e..4c124ec 100644 --- a/Deep Learning/submission_template04.py +++ b/Deep Learning/submission_template04.py @@ -1,15 +1,29 @@ import numpy as np import torch from torch import nn -from torch.nn import functional as F +import torch.nn.functional as F class ConvNet(nn.Module): def __init__(self): - ... + super().__init__() - - def forward(self, x): - ... + # 定义网络层 + self.conv1 = nn.Conv2d(in_channels=3, out_channels=16, kernel_size=(5, 5)) + self.pool1 = nn.MaxPool2d(kernel_size=2) + self.conv2 = nn.Conv2d(in_channels=16, out_channels=32, kernel_size=(3, 3)) + self.pool2 = nn.MaxPool2d(kernel_size=2) + self.flatten = nn.Flatten() + self.fc1 = nn.Linear(32 * 6 * 6, 256) + self.dropout = nn.Dropout(0.3) + self.fc2 = nn.Linear(256, 10) -def create_model(): - return ConvNet() + def forward(self, x): + x = F.relu(self.conv1(x)) + x = self.pool1(x) + x = F.relu(self.conv2(x)) + x = self.pool2(x) + x = self.flatten(x) + x = F.relu(self.fc1(x)) + x = self.dropout(x) + x = self.fc2(x) + return x