Pytorch-UNet icon indicating copy to clipboard operation
Pytorch-UNet copied to clipboard

Published 20 hours ago • verified publisher icon milesial

Run the model in CPU

Open akote123 opened this issue 6 months ago • 0 comments

I want to run the unet model in cpu. I have tried net = torch.hub.load('milesial/Pytorch-UNet', 'unet_carvana', pretrained=True, scale=0.5) But I am getting error as map this to cpu when I try to add mappings in torch.hub.load argument it says the argument is not there. So how can I run this model in CPU for inference

akote123 avatar Aug 07 '24 12:08 akote123