diff --git a/train.py b/train.py index 445f3d7652920685e833906dda758a0392db3d49..44b2ccbc46c87966461001d4a1086e0f16bbc48a 100644 --- 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()