RuntimeError: dimension out of range (expected to be in range of [-2, 1], but got 2) about cornernet

[lianbobo]

CornerNet/models/py_utils/kp_utils.py
Above line 193 ( mask = mask.unsqueeze(1) + mask.unsqueeze(2) ) add a line of code：tag_mean = tag_mean.unsqueeze(0)

[ekremcet]

CornerNet/models/py_utils/kp_utils.py
Above line 193 ( mask = mask.unsqueeze(1) + mask.unsqueeze(2) ) add a line of code：tag_mean = tag_mean.unsqueeze(0)

This solved the problem. Thanks !

[duanqipeng]

CornerNet/models/py_utils/kp_utils.py
Above line 193 ( mask = mask.unsqueeze(1) + mask.unsqueeze(2) ) add a line of code：tag_mean = tag_mean.unsqueeze(0)

This solved the problem. Thanks !

hey! i got the same error as u meet,then i use the method to solve the problem,but get another problem

IndexError: The shape of the mask [2, 128, 128] at index 0does not match the shape of the indexed tensor [1, 2, 2, 128] at index 0
i have 4 gpu and use bachsize=6,chunksize=[1,1,2,2]