From: Francois Fleuret Date: Thu, 18 Aug 2022 06:15:28 +0000 (+0200) Subject: Update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=87d376ff7929347865b199d05d003ab3b168f249;p=pytorch.git Update. --- diff --git a/minidiffusion.py b/minidiffusion.py index e7be8c1..066cbbb 100755 --- a/minidiffusion.py +++ b/minidiffusion.py @@ -66,12 +66,14 @@ def sample_mnist(nb): return result samplers = { - 'gaussian_mixture': sample_gaussian_mixture, - 'ramp': sample_ramp, - 'two_discs': sample_two_discs, - 'disc_grid': sample_disc_grid, - 'spiral': sample_spiral, - 'mnist': sample_mnist, + f.__name__.removeprefix('sample_') : f for f in [ + sample_gaussian_mixture, + sample_ramp, + sample_two_discs, + sample_disc_grid, + sample_spiral, + sample_mnist, + ] } ###################################################################### diff --git a/minidiffusion_gaussian_mixture.pdf b/minidiffusion_gaussian_mixture.pdf index 04d9b3d..4960e3e 100644 Binary files a/minidiffusion_gaussian_mixture.pdf and b/minidiffusion_gaussian_mixture.pdf differ diff --git a/minidiffusion_ramp.pdf b/minidiffusion_ramp.pdf index 17c11b8..8fc2c68 100644 Binary files a/minidiffusion_ramp.pdf and b/minidiffusion_ramp.pdf differ