X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=minidiffusion.py;fp=minidiffusion.py;h=ad1cda07d6e3854bedc54e77ea174f0e09ce05e4;hb=b740a738f11ec566e99ac9c4f674119e7e9428b7;hp=cbdb1425b090d7f13b33eaa6779e8a9177821d70;hpb=bc937c74ad8cbeede2c2ae97cda72eaa3e9bb4f3;p=pytorch.git

diff --git a/minidiffusion.py b/minidiffusion.py
index cbdb142..ad1cda0 100755
--- a/minidiffusion.py
+++ b/minidiffusion.py
@@ -81,6 +81,8 @@ ax.hist(d, 25, (-1, 1), histtype = 'stepfilled', color = 'lightblue', density =
 d = x.flatten().detach().numpy()
 ax.hist(d, 25, (-1, 1), histtype = 'step', color = 'red', density = True, label = 'Synthesis')
 
+ax.legend(frameon = False, loc = 2)
+
 filename = 'diffusion.pdf'
 fig.savefig(filename, bbox_inches='tight')