image 1/2 data/samples/bus.jpg: Traceback (most recent call last):


 "detect.py", line 179, in <module>

detect()


 "detect.py", line 91, in detect

pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres, classes=opt.classes, agnostic=opt.agnostic_nms)


 "/root/private/yolov3/utils/utils.py", line 557, in non_max_suppression

i = torchvision.ops.boxes.nms(boxes, scores, iou_thres)


 "/usr/local/lib/python3.6/dist-packages/torchvision/ops/boxes.py", line 31, in nms

return torch.ops.torchvision.nms(boxes, scores, iou_threshold)
RuntimeError: Not compiled with GPU support (nms at /tmp/pip-req-build-dkbjnmj4/torchvision/csrc/nms.h:20)

> Traceback (most recent call last):


 "train.py", line 38, in <module>

main(opt)


 "train.py", line 24, in main

single_main(opt)


 "/home/ruiy/disk_86/OpenNMT-py/onmt/train_single.py", line 128, in main

opt.valid_steps)


 "/home/ruiy/disk_86/OpenNMT-py/onmt/trainer.py", line 177, in train

report_stats)


 "/home/ruiy/disk_86/OpenNMT-py/onmt/trainer.py", line 290, in _gradient_accumulation

self.model(src, tgt, src_lengths, dec_state)


 "/home/ruiy/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

result = self.forward(*input, **kwargs)


 "/home/ruiy/disk_86/OpenNMT-py/onmt/models/model.py", line 45, in forward

enc_final, memory_bank = self.encoder(src, lengths)


 "/home/ruiy/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

result = self.forward(*input, **kwargs)


 "/home/ruiy/disk_86/OpenNMT-py/onmt/encoders/transformer.py", line 112, in forward

out = self.transformer[i](out, mask)


 "/home/ruiy/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

result = self.forward(*input, **kwargs)


 "/home/ruiy/disk_86/OpenNMT-py/onmt/encoders/transformer.py", line 48, in forward

input_norm = self.layer_norm(inputs)


 "/home/ruiy/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

result = self.forward(*input, **kwargs)


 "/home/ruiy/disk_86/OpenNMT-py/onmt/modules/util_class.py", line 20, in forward

return self.a_2 * (x - mean) / (std + self.eps) + self.b_2
RuntimeError: The size of tensor a (512) must match the size of tensor b (300) at non-singleton dimension 2
