专栏名称: 爱可可-爱生活
知名互联网资讯博主 北邮PRIS模式识别实验室陈老师
今天看啥  ›  专栏  ›  爱可可-爱生活

【基于 PyTorch 的分布式 VAE(变分自编码器)训练器的-20241012130029

爱可可-爱生活  · 微博  · AI  · 2024-10-12 13:00
    

文章预览

2024-10-12 13:00 本条微博链接 【基于 PyTorch 的分布式 VAE(变分自编码器)训练器的实现,该训练器用于训练潜在扩散模型(如稳定扩散模型、FLUX、SORA 等)中的 VAE,并包含了一些修改和优化,如固定方差、LPIPS 损失、低通滤波器重建损失等,以提高训练稳定性和性能】'Train VAE like a boss - My attempt to write distributed VAE trainer' GitHub: github.com/cloneofsimo/vqgan-training #VAE训练# #人工智能# #图像处理# ………………………………

原文地址:访问原文地址
快照地址: 访问文章快照
总结与预览地址:访问总结与预览