发现原因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)
发现原因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)