Traceback (most recent call last):


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

train(hyp, opt, device, tb_writer)


 "train.py", line 308, in train

save_dir=log_dir)


 "/root/yolo_v4_u3_final/PyTorch_YOLOv4-master/test.py", line 128, in test

output = non_max_suppression(inf_out, conf_thres=conf_thres, iou_thres=iou_thres, merge=merge)


 "/root/yolo_v4_u3_final/PyTorch_YOLOv4-master/utils/general.py", line 619, 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 40, in nms

return torch.ops.torchvision.nms(boxes, scores, iou_threshold)
RuntimeError: CUDA out of memory. Tried to allocate 4.97 GiB (GPU 0; 10.76 GiB total capacity; 928.81 MiB already ated; 3.26 GiB free; 6.51 GiB reserved in total by PyTorch)

2015-01-05 12:25:15,200 [1420457115] Critical : Back trace of it: Traceback (most recent call last):

 "/usr/local/lib/python2.6/dist-packages/Shinken-2.0.3-py2.6.egg/shinken/daemons/schedulerdaemon.py", line 479, in main
self.do_mainloop()

 "/usr/local/lib/python2.6/dist-packages/Shinken-2.0.3-py2.6.egg/shinken/daemon.py", line 322, in do_mainloop
self.do_loop_turn()

 "/usr/local/lib/python2.6/dist-packages/Shinken-2.0.3-py2.6.egg/shinken/daemons/schedulerdaemon.py", line 331, in do_loop_turn
self.setup_new_conf()

 "/usr/local/lib/python2.6/dist-packages/Shinken-2.0.3-py2.6.egg/shinken/daemons/schedulerdaemon.py", line 347, in setup_new_conf
conf = cPickle.loads(conf_raw)UnpicklingError: NEWOBJ class argument isn't a type object