ddpm-ema-anime-v2-128

模型描述

这个扩散模型是使用 ? Diffusers 库在 huggan/selfie2anime 数据集上进行训练的。

使用方法和限制

from diffusers import DDPMPipeline

model_id = "mrm8488/ddpm-ema-anime-v2-128"

# load model and scheduler
pipeline = DDPMPipeline.from_pretrained(model_id)

# run pipeline in inference 
image = pipeline()["sample"]

image[0].save("anime_face_128.png")

[TODO: 提供潜在问题和可能的修复示例]

训练数据

[TODO: 描述用于训练模型的数据]

训练超参数

训练过程中使用了以下超参数：

• 学习率：0.0001
• 训练批大小：2
• 评估批大小：4
• 梯度累积步数：1
• 优化器：AdamW，beta 值为 (0.95, 0.999)，权重衰减为 1e-06， epsilon 为 1e-08
• 学习率调度器：余弦
• 学习率预热步数：500
• ema_inv_gamma：1.0
• ema_inv_gamma：0.75
• ema_inv_gamma：0.9999
• 混合精度：fp16

训练结果

? TensorBoard logs

由 Manuel Romero/@mrm8488 创建，并得到了 Q Blocks 的支持