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

PixArt-Sigma basic training/inference script #639

Merged
merged 47 commits into from
Sep 24, 2024

Conversation

zhtmike
Copy link
Collaborator

@zhtmike zhtmike commented Aug 21, 2024

What does this PR do?

Fixes # (issue)

Adds # (feature)

Before submitting

  • This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
  • Did you read the contributor guideline?
  • Did you make sure to update the documentation with your changes? E.g. record bug fixes or new features in What's New. Here are the
    documentation guidelines
  • Did you build and run the code without any errors?
  • Did you report the running environment (NPU type/MS version) and performance in the doc? (better record it for data loading, model inference, or training tasks)
  • Did you write any new necessary tests?

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.

@xxx

@zhtmike zhtmike requested a review from geniuspatrick as a code owner August 22, 2024 10:15
@zhtmike zhtmike requested a review from vigo999 as a code owner August 27, 2024 10:46
@zhtmike zhtmike mentioned this pull request Sep 3, 2024
6 tasks
@zhtmike zhtmike changed the title PixArt-Sigma training script PixArt-Sigma basic training/inference script Sep 3, 2024

## Benchmark

### Training
Copy link
Collaborator

Choose a reason for hiding this comment

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

pls add qualititative or visual evaluation results

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

done

]
)

@ms.jit
Copy link
Collaborator

Choose a reason for hiding this comment

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

why add jit?

Copy link
Collaborator Author

@zhtmike zhtmike Sep 5, 2024

Choose a reason for hiding this comment

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

almost all mindspore optimizers implementation are using jit to accelerate training. Here it is following the same practice.


return y

@ms.jit
Copy link
Collaborator

Choose a reason for hiding this comment

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

why add jit?

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

it is faster in batch inference in pynative mode, no effect in graph mode

@CaitinZhao CaitinZhao added this pull request to the merge queue Sep 24, 2024
Merged via the queue into mindspore-lab:master with commit 2a7adcf Sep 24, 2024
3 checks passed
@zhtmike zhtmike deleted the pixart_train branch September 26, 2024 02:41
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.

6 participants