updated with pull requests, fixed logging

gorkemgoknar · Dec 21, 2020 · af3bab1 · af3bab1
1 parent 21c90f1
commit af3bab1
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/__pycache__/utils.cpython-36.pyc b/__pycache__/utils.cpython-36.pyc
diff --git a/train.py b/train.py
@@ -174,6 +174,7 @@ def train():
 
     # logging is set to INFO (resp. WARN) for main (resp. auxiliary) process. logger.info => log main process only, logger.warning => log all processes
     logging.basicConfig(level=logging.INFO if args.local_rank in [-1, 0] else logging.WARN)
+
     logger.warning("Running process %d", args.local_rank)  # This is a logger.warning: it will be printed by all distributed processes
     logger.info("Arguments: %s", pformat(args))
 

diff --git a/utils.py b/utils.py
@@ -17,10 +17,10 @@
 HF_FINETUNED_MODEL = "https://s3.amazonaws.com/models.huggingface.co/transfer-learning-chatbot/gpt_personachat_cache.tar.gz"
 
 logger = logging.getLogger(__file__)
-logger.setLevel(level=logging.DEBUG)
-mp.log_to_stderr(level=logging.DEBUG)
+logger.setLevel(level=logging.INFO)
+mp.log_to_stderr(level=logging.INFO)
 mp_logger = mp.get_logger()
-mp_logger.setLevel(level=logging.DEBUG)
+mp_logger.setLevel(level=logging.INFO)
 
 
 def download_pretrained_model():