Chris White
Results
1
comments of
Chris White
@iTarek @Djeevs @Anandh-iOS This should work. ```import torch import coremltools as ct from model import U2NET net = U2NET(3,1) net.load_state_dict(torch.load("u2net.pth", map_location=torch.device('cpu'))) net.eval() example_input = torch.rand(1, 3, 512, 512) traced_model =...