search

natanielruiz / deep-head-pose

:fire::fire: Deep Learning Head Pose Estimation using PyTorch.
Other
1.58k stars 368 forks source link

Update train_alexnet.py #127

Open Flamingo-x opened 1 year ago

Flamingo-x commented 1 year ago

I think the operation to clear the gradient is missing here