GPT. refine args print. (PaddlePaddle#701)

zzz2010 · Jul 7, 2021 · 7c76b2f · 7c76b2f
1 parent 7be95b5
commit 7c76b2f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/examples/language_model/gpt/args.py b/examples/language_model/gpt/args.py
@@ -14,6 +14,7 @@
 
 import argparse
 
+import paddle
 from paddlenlp.utils.log import logger
 
 
@@ -264,4 +265,8 @@ def parse_args(MODEL_CLASSES):
                 "The attention_probs_dropout_prob should set to 0 for accuracy checking."
             )
 
+    logger.info('{:20}:{}'.format("paddle commit id", paddle.version.commit))
+    for arg in vars(args):
+        logger.info('{:20}:{}'.format(arg, getattr(args, arg)))
+
     return args