(self, block, planes, num_blocks, stride)
| 123 | self.linear = nn.Linear(64*wide, out_dims) |
| 124 | |
| 125 | def _make_layer(self, block, planes, num_blocks, stride): |
| 126 | strides = [stride] + [1]*(num_blocks-1) |
| 127 | layers = [] |
| 128 | for stride in strides: |
| 129 | layers.append(block(self.in_planes, planes, stride, self.wide)) |
| 130 | self.in_planes = planes * self.wide * block.expansion |
| 131 | |
| 132 | return nn.Sequential(*layers) |
| 133 | |
| 134 | def forward(self, x): |
| 135 | out = F.relu(self.bn1(self.conv1(x))) |