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連休で色々動かしてみようと思います。