take batch size into account when calculating num_parleys
@@ -93,7 +93,7 @@ def main():
log_time = Timer()
print("[ training... ]")
parleys = 0
- num_parleys = opt['num_epochs'] * len(world)
+ num_parleys = opt['num_epochs'] * int(len(world) / opt['batchsize'])
best_accuracy = 0
impatience = 0
saved = False
@@ -154,5 +154,3 @@ def main():
if __name__ == '__main__':
main()
-
-
