From 243bf8ce425f11fc1743ce2cf16eec8774f11027 Mon Sep 17 00:00:00 2001 From: ChenXin Date: Mon, 23 Sep 2019 16:38:13 +0800 Subject: [PATCH] fix the bug on the trainer --- fastNLP/core/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fastNLP/core/trainer.py b/fastNLP/core/trainer.py index c331ab18..47d9edc5 100644 --- a/fastNLP/core/trainer.py +++ b/fastNLP/core/trainer.py @@ -829,12 +829,12 @@ class Trainer(object): self.best_metric_indicator = indicator_val else: if self.increase_better is True: - if indicator_val >= self.best_metric_indicator: + if indicator_val > self.best_metric_indicator: self.best_metric_indicator = indicator_val else: is_better = False else: - if indicator_val <= self.best_metric_indicator: + if indicator_val < self.best_metric_indicator: self.best_metric_indicator = indicator_val else: is_better = False