flairNLPGH-1344: Fix logging errors

Log new parameter values only for SequenceTagger instances
klasocki · Jan 11, 2020 · 97becbf · 97becbf
1 parent e3d3233
commit 97becbf
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/flair/trainers/trainer.py b/flair/trainers/trainer.py
@@ -31,6 +31,7 @@
     Result,
     store_embeddings,
 )
+from flair.models import SequenceTagger
 import random
 
 log = logging.getLogger("flair")
@@ -172,11 +173,12 @@ def train(
         log.info(f"Device: {flair.device}")
         log_line(log)
         log.info(f"Embeddings storage mode: {embeddings_storage_mode}")
-        log.info(f"Using F-score with beta: {self.model.beta}")
-        log.info(f"Weight tensor: {self.model.weights}")
-        if self.model.weights and self.model.use_crf:
-            log_line(log)
-            log.warning(f'Warning: Specified class weights will not take effect when using CRF')
+        if isinstance(self.model, SequenceTagger):
+            log.info(f"Using F-score with beta: {self.model.beta}")
+            log.info(f"Weight tensor: {self.model.weights}")
+            if self.model.weights and self.model.use_crf:
+                log_line(log)
+                log.warning(f'Warning: Specified class weights will not take effect when using CRF')
 
         # determine what splits (train, dev, test) to evaluate and log
         log_train = True if monitor_train else False