Hi,
today, I want save fasterrcnn with torch.jit.trace
like this:
import torch
import torchvision
model = torchvision.models.detection.fasterrcnn_mobilenet_v3_large_320_fpn(weights=FasterRCNN_MobileNet_V3_Large_320_FPN_Weights.DEFAULT)
data = (torch.rand(3, 300, 400), torch.rand(3, 500, 400))
torchscript_fasterrcnn = torch.jit.trace(model, data)
torch.jit.save(torchscript_fasterrcnn, 'model.pt')
But, there is a error about:
AttributeError: 'Tensor' object has no attribute 'items'
how to make it work? thanks.