[Error about finetuning on Ref-Youtube-VOS] No "dataset_name" key in the "targets"

[YOONSEOKHEO]

Hello. First of all, it is very grateful to share your wonderful work on Github.

I was trying to fine-tune your model on Ref-Youtube-VOS dataset with the backbone(video_swin_t_p4w7).
I followed the sample script you mentioned on the following link.
(./scripts/dist_train_test_ytvos.sh ytvos_dirs/video_swin_tiny pretrained_weights/video_swin_tiny_pretrained.pth --backbone video_swin_t_p4w7)

I bumped into the following error:

This error comes from this part:

That's because there is no key named "dataset_name" in the variable "target" which was defined in the getitem function.(getitem link: https://github.com/wjn922/ReferFormer/blob/9c8f237adc260c512a1c5ecfc7aee81b8282649a/datasets/ytvos.py#L166

Here is the debugging output of the target value in the error prompt..

Am I missing something??
I could not find a way to solve this error.
I am really appreciating if you give me any hints for this.

[wjn922]

Thanks for pointing this out. We have fixed the bug as follow. Please have a try now.

ReferFormer/util/misc.py

Line 542 in e40f70b

if "dataset_name" in targets[0]: