search

PacktPublishing / Modern-Computer-Vision-with-PyTorch-2E

Modern Computer Vision with PyTorch, 2E, Published by Packt
MIT License
98 stars 56 forks source link

erroe in chapter 3 #7

Open AbdelrahimKoura opened 3 months ago

AbdelrahimKoura commented 3 months ago

In the notebook Chapter03/Batch_normalization.ipynb in function def train_batch(x, y, model, opt, loss_fn): the author use the following cod optimizer.step(), optimizer.zero_grad() but must use the parameter name opt

sizhky commented 2 weeks ago

can you elaborate the code snippet? The diffs in the commit are not helpful