(self, l1=120, l2=84)
| 106 | |
| 107 | class Net(nn.Module): |
| 108 | def __init__(self, l1=120, l2=84): |
| 109 | super().__init__() |
| 110 | self.conv1 = nn.Conv2d(3, 6, 5) |
| 111 | self.pool = nn.MaxPool2d(2, 2) |
| 112 | self.conv2 = nn.Conv2d(6, 16, 5) |
| 113 | self.fc1 = nn.Linear(16 * 5 * 5, l1) |
| 114 | self.fc2 = nn.Linear(l1, l2) |
| 115 | self.fc3 = nn.Linear(l2, 10) |
| 116 | |
| 117 | def forward(self, x): |
| 118 | x = self.pool(F.relu(self.conv1(x))) |
nothing calls this directly
no outgoing calls
no test coverage detected