Mobile optimization error - PyTorch Forums
discuss.pytorch.org › t › mobile-optimization-errorNov 02, 2020 · I am trying to execute the retinanet model included in torchvision on an android mobile with Pytorch Mobile. When using the following snippet : import torch import torchvision from torch.utils.mobile_optimizer import optimize_for_mobile model = torchvision.models.detection.retinanet_resnet50_fpn(pretrained=True) model.eval() traced_script_module = torch.jit.script(model) traced_script_module ...