Searched refs:in_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)
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)
30 nn.Conv2d(in_channels, out_channels, kernel_size=1, stride=stride[0], padding=0),
|