Open AdrianCiotinga opened 9 months ago
Quick fix for memory leak in forward function of YOLOv5TorchObjectDetector that is caused by multiplying a tensor by a view of itself and reassigning that to a view of the original tensor.
Quick fix for memory leak in forward function of YOLOv5TorchObjectDetector that is caused by multiplying a tensor by a view of itself and reassigning that to a view of the original tensor.