X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=minidiffusion.py;h=e7be8c1c8651a3cc4ee4a578586e2e4dd3c29bcf;hb=2ce3bbc02bfc66e0b3366d34d8c3028d1d99b13d;hp=c88765ce2cbe0a92647081a1a60a3b4954a33e56;hpb=b19c2b7ddf3e4db73d422c3c7e6c4371f9d6e657;p=pytorch.git

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)