MichiganCOG / ViP

Video Platform for Action Recognition and Object Detection in Pytorch
MIT License
219 stars 37 forks source link

Scaling gradients when params requires_grad is False #7

Closed natlouis closed 5 years ago

natlouis commented 5 years ago

If a network layer is frozen, the pseudo batch loop code crashes when it tries to scale the gradient.