Enable logging operations in model file to show model initialization …

…configs
stanfordnlp · Jan 9, 2024 · d4d1f33 · d4d1f33
1 parent 39f9c98
commit d4d1f33
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/stanza/models/lemma_classifier/model.py b/stanza/models/lemma_classifier/model.py
@@ -1,6 +1,7 @@
 import torch
 import torch.nn as nn
 import os
+import logging
 from torch.nn.utils.rnn import pad_sequence, pack_padded_sequence, pad_packed_sequence
 from stanza.models.common.char_model import CharacterModel, CharacterLanguageModel
 from typing import List, Tuple
@@ -10,6 +11,9 @@
 from stanza.models.lemma_classifier.base_model import LemmaClassifier
 from stanza.models.lemma_classifier.constants import ModelType
 
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+
+
 class LemmaClassifierLSTM(LemmaClassifier):
     """
     Model architecture: