when i was trianing this tiny psg dataset ， it show following error

[qqqqhhqq]

File "/home/new702/disk3/QH/Workstation/OpenPSG/openpsg/models/relation_heads/psgtr_head.py", line 475, in loss_single gt_masks_list, img_metas, gt_bboxes_ignore_list) File "/home/new702/disk3/QH/Workstation/OpenPSG/openpsg/models/relation_heads/psgtr_head.py", line 629, in get_targets gt_bboxes_ignore_list) File "/home/new702/disk3/envs/openpsg/lib/python3.7/site-packages/mmdet/core/utils/misc.py", line 30, in multi_apply return tuple(map(list, zip(*map_results))) File "/home/new702/disk3/QH/Workstation/OpenPSG/openpsg/models/relation_heads/psgtr_head.py", line 721, in _get_target_single gt_sub_bboxes = torch.vstack(gt_sub_bboxes).type_as(gt_bboxes).reshape( RuntimeError: vstack expects a non-empty TensorList

it may be an error from a ground truth annotation ?

[Jingkang50]

Thanks for helping to answer questions! For your question, make sure you uncomment this https://github.com/Jingkang50/OpenPSG/blob/ae418a15f6c0d558b665ffaca3ec0dd52e0f6f24/openpsg/datasets/psg.py#L79

[qqqqhhqq]

Thank you for your answer. The program can run normally. Thank you again.

[Chandu-4444]

Where can I get that tiny_psg dataset?

[Jingkang50]

Hi @Chandu-4444, please check our latest news