Searched refs:RestNetDownBlock (Results 1 – 1 of 1) sorted by relevance
| /OK3568_Linux_fs/external/rknn-toolkit2/examples/pytorch/resnet18_export_onnx/ |
| H A D | test.py | 22 class RestNetDownBlock(nn.Module): class
24 super(RestNetDownBlock, self).__init__()
54 self.layer2 = nn.Sequential(RestNetDownBlock(64, 128, [2, 1]),
57 self.layer3 = nn.Sequential(RestNetDownBlock(128, 256, [2, 1]),
60 self.layer4 = nn.Sequential(RestNetDownBlock(256, 512, [2, 1]),
|