RuntimeError: ожидаемый объект скалярного типа Double, но получил скалярный тип Float для аргумента # 2 'other' при вызове _th_max

Файл /home/jake/venv/lib/python3.7/site-packages/torchvision/models/detection/rpn.py, строка 274, в assign_targets_to_anchors match_quality_matrix = self.box_similarity (gt_boxes, anchors_per_image) File / home / jav /lib/python3.7/site-packages/torchvision/ops/boxes.py, строка 153, в box_iou lt = torch.max (box1 [:, None,: 2], box2 [:,: 2]) # [ N, M, 2] RuntimeError: ожидаемый объект скалярного типа Double, но получил скалярный тип Float для аргумента № 2 'other' при вызове _th_max

получил следующую ошибку и вот тип 

annotations-> [{'boxes': tensor([[131.5124, 169.4232, 165.8235, 209.7217],
        [157.2659, 185.7604, 199.0850, 210.4478],
        [184.4726, 171.3594, 211.5178, 210.3267],
        [170.5867, 154.1751, 211.7600, 188.7858],
        [162.8364, 156.2324, 182.1313, 186.7285],
        [131.2702, 149.8185, 149.4349, 172.3276],
        [192.0614, 140.5002, 208.4500, 155.2642],
        [165.2583, 140.5002, 188.2670, 158.0476],
        [169.5371, 131.9080, 193.3531, 151.1497],
        [131.1895, 133.2392, 140.1507, 156.3534]], device='cuda:0',
       dtype=torch.float64), 'labels': tensor([1, 1, 1, 1, 1, 1, 1, 1, 1, 1], device='cuda:0'), 'area': tensor([1382.6857, 1032.4053, 1053.8795, 1425.0382,  588.4218,  408.8705,
         241.9614,  403.7413,  458.2574,  207.1317], device='cuda:0'), 'iscrowd': tensor([0, 0, 0, 0, 0, 0, 0, 0, 0, 0], device='cuda:0')}]

Я думаю, что тип ящиков - float64. это вызывает проблему, как исправить, чтобы удвоить?


python pytorch torch torchvision tensor
person Community    schedule 01.07.2020    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Трудно сказать без контекста.

Но исправление состоит в том, чтобы вызвать mytensor = mytensor.double() для каждого из ваших тензоров, а также mymodel = mymodel.double() для преобразования ваших данных и модели в один и тот же двойной тип. В качестве альтернативы вы можете заменить .double() на .float() для типа с плавающей запятой. Оба подхода должны работать нормально

person Alka    schedule 01.07.2020