You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Maintain an epoch variable in class as self.epoch to track how many epochs are done till now. Take e as an argument in trainer function in class train. Use e in outer loop -- for times to iterate on the dataset. Add that number to self. epoch. Also, take display step as an argument.
Set number of Epoch and Display step from the trainer function
The text was updated successfully, but these errors were encountered: