tgxs002 / CORA

A DETR-style framework for open-vocabulary detection (OVD). CVPR 2023
Apache License 2.0
166 stars 14 forks source link

New category is None #23

Open ZitengXue opened 9 months ago

ZitengXue commented 9 months ago

Hi! authors, thanks for your great job! I tried to train the code with the order: bash configs/COCO/R50_dab_ovd_3enc_apm128_splcls0.2_relabel_noinit.sh RN50 2 local and set batch size = 2 When running to the second iteration, the new_category in Classifier (line 131) is None and the training is stoped. How can I fix it?