Linear(84, classes) # 子模組拼接def forward(self, x): out = F.relu(self.conv1(x)) out = F.max_pool2d(out, 2) out = F.relu(self.conv2(out)) ...
確定! 回上一頁