A error from FasterRCNN

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.