seanzhuh / SeqTR

SeqTR: A Simple yet Universal Network for Visual Grounding
https://arxiv.org/abs/2203.16265
128 stars 14 forks source link

size mismatch for lan_enc.embedding.weight #35

Open Hasanmog opened 4 months ago

Hasanmog commented 4 months ago

Hello , Thanks for your work. I was trying to test this model on a dataset by using this command : python tools/test.py configs/seqtr/detection/seqtr_det_refcoco-unc.py --load-from data/weights/weight_2/det_best.pth --cfg-options ema=True , I'm using the RefCOCO (pre-trained + fine-tuned SeqTR detection) weight file for this. But I keep getting this error :

RuntimeError: Error(s) in loading state_dict for SeqTR:
        size mismatch for lan_enc.embedding.weight: copying a param with shape torch.Size([10344, 300]) from checkpoint, the shape in current model is torch.Size([27, 300]).

Any ideas ?