Code Error Issue

[jarch-ma]

Thank you for your excellent work!
However, I encountered some issues recently while trying to reproduce the experiments. Specifically, on the COCO dataset (e.g., fold 1, 701/1000) or the LVIS dataset (e.g., fold 1, 4/1000) , the code consistently crashes after inferring a certain number of images. The error message is as follows:

Traceback (most recent call last):
File "main_eval.py", line 118, in
test_miou, test_fb_iou = test(GFSAM, dataloader_test, args=args)
File "main_eval.py", line 43, in test
pred_mask, _ = GFSAM.predict()
File "/code/GF-SAM/matcher/GFSAM.py", line 134, in predict
return pred_masks, (coord_xy, selected_points)
UnboundLocalError: local variable 'selected_points' referenced before assignment

Thanks for your help!