search

lutzroeder / netron

Visualizer for neural network, deep learning and machine learning models
https://netron.app
MIT License
28.31k stars 2.8k forks source link

TorchScript server #990

Open lutzroeder opened 2 years ago

lutzroeder commented 2 years ago 
import torch
import torchvision
import torch.utils.tensorboard
model = torchvision.models.detection.fasterrcnn_resnet50_fpn()
script = torch.jit.script(model)
script.save('fasterrcnn_resnet50_fpn.pt')
with torch.utils.tensorboard.SummaryWriter('log') as writer:
    writer.add_graph(script, ())

fasterrcnn_resnet50_fpn.pt.zip

HLearning commented 1 year ago

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