From 5eacc141ac6e515ef8f1c04d714433bf26f5e318 Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Thu, 16 Jan 2020 10:33:52 +0100 Subject: [PATCH] typo --- pts/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pts/trainer.py b/pts/trainer.py index 44a1065..4f3ca71 100644 --- a/pts/trainer.py +++ b/pts/trainer.py @@ -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()