From: Francois Fleuret Date: Wed, 17 Aug 2022 08:10:18 +0000 (+0200) Subject: Cosmetics + added the figures. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=2ce3bbc02bfc66e0b3366d34d8c3028d1d99b13d;p=pytorch.git Cosmetics + added the figures. --- diff --git a/minidiffusion.py b/minidiffusion.py index c88765c..e7be8c1 100755 --- a/minidiffusion.py +++ b/minidiffusion.py @@ -313,7 +313,7 @@ if train_input.dim() == 2 and train_input.size(1) == 1: ax.legend(frameon = False, loc = 2) - filename = f'diffusion_{args.data}.pdf' + filename = f'minidiffusion_{args.data}.pdf' print(f'saving {filename}') fig.savefig(filename, bbox_inches='tight') @@ -350,7 +350,7 @@ elif train_input.dim() == 2 and train_input.size(1) == 2: ax.legend(frameon = False, loc = 2) - filename = f'diffusion_{args.data}.pdf' + filename = f'minidiffusion_{args.data}.pdf' print(f'saving {filename}') fig.savefig(filename, bbox_inches='tight') @@ -375,7 +375,7 @@ elif train_input.dim() == 4: result = 1 - torch.cat((t, x), 2) / 255 - filename = f'diffusion_{args.data}.png' + filename = f'minidiffusion_{args.data}.png' print(f'saving {filename}') torchvision.utils.save_image(result, filename) diff --git a/minidiffusion_disc_grid.pdf b/minidiffusion_disc_grid.pdf new file mode 100644 index 0000000..a97fcf3 Binary files /dev/null and b/minidiffusion_disc_grid.pdf differ diff --git a/minidiffusion_gaussian_mixture.pdf b/minidiffusion_gaussian_mixture.pdf new file mode 100644 index 0000000..04d9b3d Binary files /dev/null and b/minidiffusion_gaussian_mixture.pdf differ diff --git a/minidiffusion_mnist.png b/minidiffusion_mnist.png new file mode 100644 index 0000000..648e7f3 Binary files /dev/null and b/minidiffusion_mnist.png differ diff --git a/minidiffusion_ramp.pdf b/minidiffusion_ramp.pdf new file mode 100644 index 0000000..17c11b8 Binary files /dev/null and b/minidiffusion_ramp.pdf differ diff --git a/minidiffusion_spiral.pdf b/minidiffusion_spiral.pdf new file mode 100644 index 0000000..0a47a77 Binary files /dev/null and b/minidiffusion_spiral.pdf differ diff --git a/minidiffusion_two_discs.pdf b/minidiffusion_two_discs.pdf new file mode 100644 index 0000000..4e34be5 Binary files /dev/null and b/minidiffusion_two_discs.pdf differ