らんだむな記憶

blogというものを体験してみようか!的なー

拡散モデル

DDPM

  • https://github.com/cloneofsimo/minDiffusion を使用。
  • MNIST, KMNIST 共に a2-highgpu-1g で 8494MiB 程度の VRAM を使用。
    • batch_size < 8 で、VRAM の使用量が 1GB 程度まで落ちる。batch_size = 8 で 6GB 程度。何故急に増えるかは未調査。
  • 1 epoch あたり、約 55 秒。