Browse Source

fix the bug on the trainer

tags/v0.4.10
ChenXin 5 years ago
parent
commit
243bf8ce42
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      fastNLP/core/trainer.py

+ 2
- 2
fastNLP/core/trainer.py View File

@@ -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


Loading…
Cancel
Save