GH-48: Set train loss to current loss per default

flairNLP · Sep 28, 2018 · 03258f3 · 03258f3
1 parent a2975c2
commit 03258f3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/flair/trainers/text_classification_trainer.py b/flair/trainers/text_classification_trainer.py
@@ -122,7 +122,8 @@ def train(self,
                 log.info("EPOCH {0}: lr {1:.4f} - bad epochs {2}".format(epoch + 1, learning_rate, scheduler.num_bad_epochs))
 
                 dev_metric = train_metric = None
-                dev_loss = train_loss = '_'
+                dev_loss = '_'
+                train_loss = current_loss
 
                 if eval_on_train:
                     train_metric, train_loss = self._calculate_evaluation_results_for(