search

Royalvice / DocDiff

ACM Multimedia 2023: DocDiff: Document Enhancement via Residual Diffusion Models. Also contains 1597 red seals in Chinese scenes, along with their corresponding binary masks.
https://www.aibupt.com/
MIT License
234 stars 24 forks source link

什么时候开发多卡训练啊，扩散的多卡训练怎么搞 #37

Open monster-star opened 2 months ago

Royalvice commented 2 weeks ago

目前没有额外的精力开发多卡了，可以用diffusers和pytorch lightning改写