import torch.nn as nn
import torch.nn.functional as F
classModel(nn.Module):def__init__(self):super(Model, self).__init__()
self.conv1 = nn.Conv2d(1,20,5)# submodule: Conv2d,子模块1
self.conv2 = nn.Conv2d(20,20,5)# 子模块2defforward(self, x):
x = F.relu(self.conv1(x))return F.relu(self.conv2(x))