我正在尝试为我的神经网络实现一个自定义数据集。但是在运行前向函数时得到了这个错误。代码如下。#thrid place
在该文件中scalar type Double but got scalar type Float for argument #2 'mat2' in call to _th_mm
似乎我应该将dataset中数字的)changing,在第四
### Create / load model
model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=True)in_features = model.roi_heads.box_predictor.cls_score.in_features
# replace the pre-tr