From: Francois Fleuret <francois@fleuret.org>
Date: Thu, 11 Aug 2022 20:52:34 +0000 (+0200)
Subject: Update.
X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=commitdiff_plain;h=b740a738f11ec566e99ac9c4f674119e7e9428b7

Update.
---

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')