cover do_detect function with torch.no_grad(): - Githubissues

Tianxiaomo / pytorch-YOLOv4

PyTorch ,ONNX and TensorRT implementation of YOLOv4

Apache License 2.0

4.47k stars 1.49k forks source link

cover do_detect function with torch.no_grad(): #434

Closed developer0hye closed 2 years ago

developer0hye commented 3 years ago

use less memory for inference