mindspore-lab / mindone

one for all, Optimal generator with No Exception
https://mindspore-lab.github.io/mindone/
Apache License 2.0
368 stars 74 forks source link

update pku vae 1.2 training #713

Closed Songyuanwei closed 3 weeks ago

Songyuanwei commented 1 month ago

What does this PR do?

  1. add vae 1.2 training.
  2. fix loss scale bug.

Fixes # (issue)

Adds # (feature)

Before submitting

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