update train.py

manogna-s · Nov 5, 2020 · 163fcb8 · 163fcb8
1 parent 237d9e5
commit 163fcb8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train.py b/train.py
@@ -234,6 +234,7 @@ def train(args, model):
 
     if args.local_rank in [-1, 0]:
         writer.close()
+    logger.info("Best Accuracy: \t%f" % best_acc)
     logger.info("End Training!")