search

mindspore-lab / mindone

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

[Open-Sora-Plan-v1.2.0]: VAE training update #731

Closed wtomin closed 2 weeks ago

wtomin commented 2 weeks ago

Support training causalvae with full precision while using recompute.

The training step time is 5.45s. The SSIM and PSNR scores are 0.88 and 29.28. The loss is converging normally for over 40k steps.