Fixed memory leak in forward function

pooya-mohammadi / yolov5-gradcam

Visualizing Yolov5's layers using GradCam

MIT License

281 stars 47 forks source link

Fixed memory leak in forward function #36

Open AdrianCiotinga opened 9 months ago

AdrianCiotinga commented 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.