Fix to_static args

examples/machine_translation/transformer/train.py

@@ -1,3 +1,17 @@
+# Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 import os
 import time
 
@@ -86,11 +100,6 @@ def parse_args():
                         type=str,
                         choices=['true', 'false', 'True', 'False'],
                         help="Whether to use amp to train Transformer. ")
-    parser.add_argument("--to_static",
-                        default=None,
-                        type=str,
-                        choices=['true', 'false', 'True', 'False'],
-                        help="Whether use to_static to train Transformer. ")
     parser.add_argument(
         "--amp_level",
         default=None,
@@ -106,6 +115,9 @@ def parse_args():
         help=
         'The option of profiler, which should be in format \"key1=value1;key2=value2;key3=value3\".'
     )
+    parser.add_argument("--to_static",
+                        action="store_true",
+                        help="Whether to_static to train Transformer. ")
     args = parser.parse_args()
     return args
 
@@ -388,12 +400,7 @@ def do_train(args):
     args.unk_token = ARGS.unk_token
     args.bos_token = ARGS.bos_token
     args.eos_token = ARGS.eos_token
-    if ARGS.to_static:
-        ARGS.to_static = ARGS.to_static.lower()
-        if ARGS.to_static == "true":
-            args.to_static = True
-        else:
-            args.to_static = False
+    args.to_static = ARGS.to_static
     pprint(args)
 
     args.profiler_options = ARGS.profiler_options

tests/test_tipc/configs/transformer/base/train_infer_python.txt

@@ -17,7 +17,7 @@ norm_train:../examples/machine_translation/transformer/train.py --config ../exam
 pact_train:null
 fpgm_train:null
 distill_train:null
-to_static_train:--to_static=True
+to_static_train:--to_static
 null:null
 ##
 ===========================eval_params=========================== 

tests/test_tipc/test_train_inference_python.sh

@@ -1,4 +1,19 @@
 #!/bin/bash
+
+# Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
+# 
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 source test_tipc/common_func.sh
 
 FILENAME=$1
@@ -296,7 +311,7 @@ else
                     run_train=${distill_trainer}
                     run_export=${distill_export}
                 # In case of @to_static, we re-used norm_traier,
-                # but append "--to_static=True" for config
+                # but append "--to_static" for config
                 # to trigger "apply_to_static" logic in 'train.py'
                 elif [ ${trainer} = "${to_static_key}" ]; then
                     run_train="${norm_trainer}  ${to_static_trainer}"