keypoint_rcnn_training_pytorch icon indicating copy to clipboard operation
keypoint_rcnn_training_pytorch copied to clipboard

I met a error when train with this code.

Open xiaoLiu0531 opened this issue 3 years ago • 0 comments

I used glue datasets and custom datasets ,but met the same question: File "D:/program/keypoint_rcnn_training_pytorch-main/trainer.py", line 164, in train_one_epoch(model, optimizer, data_loader_train, device, epoch, print_freq=1000) File "D:\program\keypoint_rcnn_training_pytorch-main\engine.py", line 31, in train_one_epoch loss_dict = model(images, targets) File "E:\python38\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl return forward_call(*input, **kwargs) File "E:\python38\lib\site-packages\torchvision\models\detection\generalized_rcnn.py", line 99, in forward detections, detector_losses = self.roi_heads(features, proposals, images.image_sizes, targets) File "E:\python38\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl return forward_call(*input, **kwargs) File "E:\python38\lib\site-packages\torchvision\models\detection\roi_heads.py", line 740, in forward assert t["labels"].dtype == torch.int64, "target labels must of int64 type" KeyError: 'labels'

xiaoLiu0531 avatar May 20 '22 13:05 xiaoLiu0531