minDiffusion
1.0.0

該教育存儲庫的目標是使用Pytorch提供獨立的,簡約的擴散模型實現。
擴散模型的許多實現可能會有些壓倒性。在這裡, superminddpm :在200行代碼下,與Pytorch完全自我包含的DDPM實現是一個很好的起點,對於那些想要開始降級擴散模型的人而無需花費時間來詳細信息。
簡單地:
$ python superminddpm.py
上面的腳本是獨立的。 (當然,您需要安裝pytorch和torchvision。最新版本就足夠了。我們不使用任何尖端功能。)
如果要使用更多重構代碼,則運行CIFAR10數據集:
$ python train_cifar10.py

上述結果大約需要2個小時的單個GPU培訓。產生了前8個圖像,底部8是地面真相。
這是另一個例子,接受了100個時期的訓練(大約1.5小時)

目前有:
戒酒