This commit is contained in:
Dr. Kashif Rasul
2020-01-16 10:33:52 +01:00
parent 00d790a66d
commit 5eacc141ac
+1 -1
View File
@@ -64,7 +64,7 @@ class Trainer:
optimizer.step()
if self.num_batches_per_epoch == batch_no:
break()
break
# mark epoch end time and log time cost of current epoch
toc = time.time()