Why is the output of the model has no grads?

eriklindernoren / PyTorch-YOLOv3

Minimal PyTorch implementation of YOLOv3

GNU General Public License v3.0

7.32k stars 2.63k forks source link

Why is the output of the model has no grads? #260

Open wzjin2017 opened 5 years ago

wzjin2017 commented 5 years ago

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?

zzh-z commented 5 years ago

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

Flova commented 3 years ago

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