From a0b34ef87a66c0dfb318e6b7341adaddabeadad0 Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Wed, 12 Jan 2022 14:36:32 -0800 Subject: [PATCH] [examples/flax/language-modeling] set loglevel --- examples/flax/language-modeling/run_mlm_flax.py | 2 +- examples/flax/language-modeling/run_t5_mlm_flax.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/flax/language-modeling/run_mlm_flax.py b/examples/flax/language-modeling/run_mlm_flax.py index 9f7a36d29346..2a65b6f692da 100755 --- a/examples/flax/language-modeling/run_mlm_flax.py +++ b/examples/flax/language-modeling/run_mlm_flax.py @@ -362,7 +362,7 @@ def main(): # Setup logging logging.basicConfig( format="%(asctime)s - %(levelname)s - %(name)s - %(message)s", - level="NOTSET", + level=logging.INFO, datefmt="[%X]", ) diff --git a/examples/flax/language-modeling/run_t5_mlm_flax.py b/examples/flax/language-modeling/run_t5_mlm_flax.py index 4a66a3cd5148..2845d9ce58f0 100755 --- a/examples/flax/language-modeling/run_t5_mlm_flax.py +++ b/examples/flax/language-modeling/run_t5_mlm_flax.py @@ -492,7 +492,7 @@ def main(): # Setup logging logging.basicConfig( format="%(asctime)s - %(levelname)s - %(name)s - %(message)s", - level="NOTSET", + level=logging.INFO, datefmt="[%X]", )