PyTorch-YOLOv3 icon indicating copy to clipboard operation
PyTorch-YOLOv3 copied to clipboard

Why is the output of the model has no grads?

Open wzjin2017 opened this issue 6 years ago • 2 comments

Hi, thank you for building this pytorch version of YOLO v3. I am implementing my codes on the basis of this project. I used the darknet model in this project and the output's requires_grad is False. How can I make the output coming out the model has grad data? Could you point me to the place where I should look and modify?

wzjin2017 avatar Aug 02 '19 16:08 wzjin2017

I have met the same problem. Have you solved this?

zzh-z avatar Oct 10 '19 02:10 zzh-z

Are you in test or training mode? The model is not differentiable in test mode as it directly outputs a set of boxes.

Flova avatar Sep 14 '21 10:09 Flova