Searched refs:out_channels (Results 1 – 1 of 1) sorted by relevance
| /OK3568_Linux_fs/external/rknn-toolkit2/examples/pytorch/resnet18_export_onnx/ |
| H A D | test.py | 7 def __init__(self, in_channels, out_channels, stride): argument
9 self.conv1 = nn.Conv2d(in_channels, out_channels, kernel_size=3, stride=stride, padding=1)
10 self.bn1 = nn.BatchNorm2d(out_channels)
11 self.conv2 = nn.Conv2d(out_channels, out_channels, kernel_size=3, stride=stride, padding=1)
12 self.bn2 = nn.BatchNorm2d(out_channels)
23 def __init__(self, in_channels, out_channels, stride): argument
25 … self.conv1 = nn.Conv2d(in_channels, out_channels, kernel_size=3, stride=stride[0], padding=1)
26 self.bn1 = nn.BatchNorm2d(out_channels)
27 … self.conv2 = nn.Conv2d(out_channels, out_channels, kernel_size=3, stride=stride[1], padding=1)
28 self.bn2 = nn.BatchNorm2d(out_channels)
[all …]
|