Should backward() function be in the loop of epoch or batch?

Your explanation is correct and @albanD posted some examples in this post a while ago.

1 Like