enable the log of gradient clipping in training

7 years ago · 3f608b3b81
parent 2c63f3413f
commit 3f608b3b81
1 changed files with 3 additions and 1 deletions
--- 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()