Skip to content
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

support auto parallel in dit and largedit #551

Merged
merged 35 commits into from
Oct 10, 2024

Conversation

jeff41404
Copy link
Contributor

@jeff41404 jeff41404 commented May 23, 2024

support auto parallel in model of Dit and LargeDit, include:

  • support model of Dit and LargeDit to be trained using auto parallel in dynamic graph mode (hereinafter referred to as auto).
  • support model of Dit and LargeDit to be trained using auto parallel in dynamic to static graph mode with pir (hereinafter referred to as auto-static).
  • support hybrid parallel(DP、Sharding、MP) in model of Dit and LargeDit, PP under verification.

compare loss of manual parallel(present, hereinafter referred to as manual), auto and auto-static
loss_h_s_pir

compare loss diff of manual and auto
hand_auto_diff_pir

compare loss diff of manual and auto-static
hand_auto_static_diff

compare loss diff of auto and auto-static
auto_auto_static_diff

Copy link

paddle-bot bot commented May 23, 2024

Thanks for your contribution!

@CLAassistant
Copy link

CLAassistant commented Sep 19, 2024

CLA assistant check
All committers have signed the CLA.

Copy link
Collaborator

@jerrywgz jerrywgz left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@jeff41404
Copy link
Contributor Author

https://github.com/PaddlePaddle/PaddleMIX/tree/develop/ppdiffusers/examples/class_conditional_image_generation/DiT 在这里新增自动并行的相关文档吧

感谢提醒,已增加使用自动并行进行训练的文档说明

@nemonameless nemonameless merged commit 998ba83 into PaddlePaddle:develop Oct 10, 2024
3 checks passed
@jeff41404 jeff41404 deleted the add_auto_parallel_in_dit branch October 11, 2024 01:47
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants