Dimension out of range (expected to be in range of [-1, 0], but got 1)

[aishamina]

Traceback (most recent call last):
File "/home/lijiayi/neural-motifs-master/models/train_detector.py", line 216, in
mAp = val_epoch()
File "/home/lijiayi/neural-motifs-master/models/train_detector.py", line 171, in val_epoch
vr.append(val_batch(val_b, batch))
File "/home/lijiayi/neural-motifs-master/models/train_detector.py", line 190, in val_batch
result = detector[b]
File "/home/lijiayi/neural-motifs-master/lib/object_detector.py", line 417, in getitem
outputs = nn.parallel.parallel_apply(replicas, [batch[i] for i in range(self.num_gpus)])
File "/home/lijiayi/anaconda3/envs/motifs/lib/python3.6/site-packages/torch/nn/parallel/parallel_apply.py", line 77, in parallel_apply
raise output
File "/home/lijiayi/anaconda3/envs/motifs/lib/python3.6/site-packages/torch/nn/parallel/parallel_apply.py", line 53, in _worker
output = module(*input, **kwargs)
File "/home/lijiayi/anaconda3/envs/motifs/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/lijiayi/neural-motifs-master/lib/object_detector.py", line 311, in forward
od_box_deltas, im_sizes,
File "/home/lijiayi/neural-motifs-master/lib/object_detector.py", line 395, in nms_boxes
thresh=self.thresh,
File "/home/lijiayi/neural-motifs-master/lib/object_detector.py", line 445, in filter_det
for c_i in valid_cls.squeeze(1).cpu():
RuntimeError: Dimension out of range (expected to be in range of [-1, 0], but got 1)