demo issue

[buptzqr]

when I run the demo, I got a error,please help me,thank you.

the error information as following:
y = net(xx)
File "/home/bupt-sse3/anaconda3/lib/python3.5/site-packages/torch/nn/modules/module.py", line 489, in call
result = self.forward(*input, **kwargs)
File "/home/bupt-sse3/repulsion_loss_ssd-master/ssd.py", line 103, in forward
self.priors.type(type(x.data)) # default boxes
File "/home/bupt-sse3/repulsion_loss_ssd-master/layers/functions/detection.py", line 54, in forward
ids, count = nms(boxes, scores, self.nms_thresh, self.top_k)
ValueError: not enough values to unpack (expected 2, got 0)

[ghost]

i got the same result.

[ghost]

I changed the code as follows and worked for me.
In detection.py of line 49
from

if scores.dim() == 0:
continue
to

if scores.size(0) == 0:
continue