天天看点

libtorch 加载失败

发现原因1,bn层不支持。

self.bn1 = torch.nn.BatchNorm2d(64, momentum=0.1)

原因2:MaxPool2d不支持

 self.maxpool = torch.nn.MaxPool2d(kernel_size=3, stride=2, padding=1)