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)

[oftenliu]

No description provided.

[h72001346]

the same problem,i've met

[h72001346]

but i used cuda 9 instead

[qianzhang2018]

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

[qianzhang2018]

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

[sun201711]

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

[GalyaZalesskaya]

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!

[Glutton-zh]

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