Open lutzroeder opened 1 year 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
netron not open fasterrcnn_resnet50_fpn.pt Python Error AssertionError: targets should not be none when in training mode
fasterrcnn_resnet50_fpn.pt.zip