IndexError: tuple index out of range

[freedom6927]

when I run refcoco_train.py. It accur IndexError: tuple index out of range on the code pred_rois_feas = main_model.an_roi_aligner(normed_transformed_boxes) of train.py. Could you fix it ?

[almumujin]

I had the same problem。

[HoBeom]

M-DGT/models/MDGT_framework.py

Line 213 in 070020e

built_graph.batch_edges_index, hs, ss

also ss is not assigned.Maybe we should change it to ws.

[HoBeom]

https://github.com/pytorch/vision/blob/c4c28dff8a68fae20eb7f2a82051ef7a388353bd/torchvision/ops/poolers.py#L280

I think we should use 5 or [5, 5].

M-DGT/f30k_eval.py

Line 65 in 070020e

flags.DEFINE_list("roi_output_size", [5], "output size of the roi alignment")

[worker121]

Hello? I met the same question, when i run f30k_train.py, It accur IndexError: tuple index out of range. Is there any way to solve it?