Skip to content

Latest commit

 

History

History
22 lines (17 loc) · 1.13 KB

readme.md

File metadata and controls

22 lines (17 loc) · 1.13 KB

Unconditional Latent Diffusion Model

目录

简介

此项目为2023年系科协暑培CV课程作业,训练一个 Unconditional Latent Diffusion Model。 此项目基于 DiffusionFastForward 开发,按照原作者指定的 LICENSE 使用,特此鸣谢。

环境

详见requirements.txt

数据集

此项目采用 MNIST-M 数据集进行训练:在论文 Domain-Adversarial Training of Neural Networks 中首次使用的数据集,包含约 60000 张 28*28 的彩色手写数字图片。

清华云盘可以找到

checkpioits

本项目训练得到epoch=26-step=200000.ckpt的checkpoint,已上传至清华云盘