dist = tag_mean.unsqueeze(1) - tag_mean.unsqueeze(2) RuntimeError: Dimension out of range (expected to be in range of [-2, 1], but got 2) about cornernet-lite HOT 7 OPEN

i can work now,batch size can't be 1,do you set it the same? i set 2,now it works

from cornernet-lite.

the same problem,i've met

from cornernet-lite.

but i used cuda 9 instead

from cornernet-lite.

the same problem,i've met，do you know how to solve it now?

from cornernet-lite.

if I want to set batch_size = 1,how to solve the problem?

from cornernet-lite.

i can work now,batch size can't be 1,do you set it the same? i set 2,now it works

WHY? Why does it work?! It solves literally everything!

I had chunk_size = [1, 1, 1], batch_size = 3. And it provoked my PC reboot for CornerNet and this issue for CornerNet_Saccade.
But with chunk_size = [2] or chunk_size = [2, 2, 2] everything works :) Thanks a lot!

from cornernet-lite.

up to now, Is there an effective solution to this problem——use the bach_size == 1;

from cornernet-lite.