Cosmetics + added the figures.

diff --git a/minidiffusion.py b/minidiffusion.py
index c88765c..e7be8c1 100755 (executable)
--- 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 (file)
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 (file)
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 (file)
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 (file)
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 (file)
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 (file)
index 0000000..4e34be5
Binary files /dev/null and b/minidiffusion_two_discs.pdf differ