PyTorch-YOLOv3
PyTorch-YOLOv3 copied to clipboard
Why is the output of the model has no grads?
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?
I have met the same problem. Have you solved this?
Are you in test or training mode? The model is not differentiable in test mode as it directly outputs a set of boxes.