We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 343c109 commit 12ba2fbCopy full SHA for 12ba2fb
python/paddle/distributed/auto_parallel/intermediate/pipeline_parallel.py
@@ -383,8 +383,10 @@ def divide_list_indices(n, k):
383
sublayer_names = [name for name, _ in model.named_sublayers()]
384
split_spec_dict = split_spec
385
for key, value in split_spec_dict.items():
386
- assert key in sublayer_names, "wrong split point"
387
- assert value is SplitPoint.END, "wrong split point flag"
+ assert (
+ key in sublayer_names
388
+ ), f"wrong split layer, expected one of {sublayer_names}"
389
+ assert value is SplitPoint.END, "not supported split point at now."
390
391
if global_spec:
392
if isinstance(global_spec, str):
0 commit comments