enable the log of gradient clipping in training

3f608b3b · Yibing Liu · 2c63f341 · 3f608b3b
显示空白变更内容
内联并排

Showing with 3 addition and 1 deletion

train.py train.py +3 -1

未找到文件。
--- a/train.py
+++ b/train.py
@@ -118,7 +118,9 @@ def train():

 def main():
    print_arguments(args)
-    paddle.init(use_gpu=args.use_gpu, trainer_count=args.trainer_count)
+    paddle.init(use_gpu=args.use_gpu,
+                trainer_count=args.trainer_count,
+                log_clipping=True)
    train()