netron icon indicating copy to clipboard operation
netron copied to clipboard

TorchScript server

Open lutzroeder opened this issue 3 years ago • 1 comments

Prototype exploring to use the Python Server to load and serve TorchScript models.

Example:

import torch
import torchvision
import netron
model = torchvision.models.resnet34()
model = torch.jit.script(model)
traced_model = torch.jit.trace(model, torch.zeros([1, 3, 224, 224]), strict=True)
torch._C._jit_pass_inline(traced_model.graph)
netron.serve('resnet34', traced_model)

lutzroeder avatar Oct 09 '22 22:10 lutzroeder

netron not open fasterrcnn_resnet50_fpn.pt Python Error AssertionError: targets should not be none when in training mode

HLearning avatar Jun 19 '23 02:06 HLearning