model = torchvision.models.resnet18(pretrained=True)
x = torch.rand(1,3,224,224)
m = torch.fx.symbolic_trace(model)
m.layer1(x)
error:*** NotImplementedError
how to solve it.
model = torchvision.models.resnet18(pretrained=True)
x = torch.rand(1,3,224,224)
m = torch.fx.symbolic_trace(model)
m.layer1(x)
error:*** NotImplementedError
how to solve it.
model = torchvision.models.resnet18(pretrained=True)
x = torch.rand(1,3,224,224)
m = torch.fx.symbolic_trace(model)
m(x)
why not m.layer1(x), sometimes I only use to infer part of network.