Track validation F1 score of best model

harvardnlp · Jun 13, 2019 · 8eed323 · 8eed323
1 parent e9ef3bd
commit 8eed323
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/train.py b/train.py
@@ -211,10 +211,9 @@ def main(args):
     val_ppl, val_f1 = eval(val_data, model, 
                            samples = args.mc_samples, count_eos_ppl = args.count_eos_ppl)
     print('--------------------------------')
-    if val_f1 > best_val_f1:
-      best_val_f1 = val_f1
     if val_ppl < best_val_ppl:
       best_val_ppl = val_ppl
+      best_val_f1 = val_f1
       checkpoint = {
         'args': args.__dict__,
         'model': model.cpu(),