diff --git a/paddlenlp/trainer/training_args.py b/paddlenlp/trainer/training_args.py index 1201e50a2044..c640829d9ad2 100644 --- a/paddlenlp/trainer/training_args.py +++ b/paddlenlp/trainer/training_args.py @@ -1319,6 +1319,7 @@ def __post_init__(self): "best_unbalanced_scheduler", "enable_offload_queue", "use_dualpipev", + "forward_backward_overlap_scheduler", "enable_dynamic_shape", ]: raise ValueError( @@ -1368,6 +1369,8 @@ def __post_init__(self): "best_unbalanced_scheduler": "best_unbalanced_scheduler" in pipeline_parallel_config, "enable_offload_queue": "enable_offload_queue" in pipeline_parallel_config, "use_dualpipev": "use_dualpipev" in pipeline_parallel_config, + "forward_backward_overlap_scheduler": "forward_backward_overlap_scheduler" + in pipeline_parallel_config, "enable_dynamic_shape": "enable_dynamic_shape" in pipeline_parallel_config, } if dygraph_pp_configs["dp_comm_overlap"]: