TensorflowでDiffusionモデルを実装してみた①

急遽地元に出張があり、忙しい週を過ごしておりました、

前回までのNSTは一旦置き、大ブームのDiffusion生成モデルを実装してみました。

参考にしたページはこちら↓
https://github.com/Seachaos/Tree.Rocks/blob/main/QuickDiffusionModel/QuickDiffusionModel.ipynb


作成したコードはこちら↓
https://github.com/Kansuke-Kotsu/ML-learning/tree/main/diffusion_model

例によって、関数の部分は別ファイルにして全体を見やすくしてます。
(今回意外とめんどかった)

一旦コード自体は動きますが、学習にめちゃくちゃ時間がかかるので、特に結果はなし。

3連休で色々動かしてみようと思います。

この記事が気に入ったらサポートをしてみませんか?