-
Notifications
You must be signed in to change notification settings - Fork 3k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Enable parallel_config to use commas as delimiters. #8677
Conversation
Thanks for your contribution! |
Codecov ReportAttention: Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## develop #8677 +/- ##
===========================================
+ Coverage 55.43% 55.51% +0.07%
===========================================
Files 626 626
Lines 98070 98064 -6
===========================================
+ Hits 54366 54438 +72
+ Misses 43704 43626 -78 ☔ View full report in Codecov by Sentry. |
paddlenlp/trainer/training_args.py
Outdated
if " " in self.pipeline_parallel_config: | ||
pipeline_parallel_config = set(self.pipeline_parallel_config.split(" ")) | ||
else: | ||
if "," in self.pipeline_parallel_config: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
要不弄个函数,现在代码里面到处if else
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done,麻烦合入一下~
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
|
||
enable_release_grads = False | ||
if args.sharding_parallel_degree > 1: | ||
enable_release_grads = "enable_release_grads" in args.sharding_parallel_config |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这个地方判断咋感觉不太对呢?原来的是 enable_release_grads = (... or ...),现在的写法相当于只由 args.pipeline_parallel_config 来决定了
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
已修改,感谢指正
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
PR types
Others
PR changes
Others
Description
上述PR添加了","作为pipeline_parallel_config,sharing_parallel_config分隔符,但遗漏了一些处理,本PR对此做了补充。