error during test on phase 1

yanxp / MetaR-CNN

Meta R-CNN : Towards General Solver for Instance-level Low-shot Learning

https://yanxp.github.io/metarcnn.html

177 stars 23 forks source link

error during test on phase 1 #21

Open mandal4 opened 4 years ago

mandal4 commented 4 years ago

Thx to your released code. Unfortunately i countered an error test on phase 1 (for base class) line 332, in inds = torch.nonzero(scores[:, j] > thresh).view(-1) IndexError: index 16 is out of range for dimension 1 (of size 16)

I guess 'scores' tensor have shape such as [15,~,~...] but the code access this tensor with index 16. Is there bug in a code..?

yanxp commented 4 years ago

In the first phase, the length of mean class attentions is 15.