From: Francois Fleuret <francois@fleuret.org>
Date: Wed, 17 Aug 2022 06:14:36 +0000 (+0200)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=b19c2b7ddf3e4db73d422c3c7e6c4371f9d6e657;p=pytorch.git

Update.
---

diff --git a/minidiffusion.py b/minidiffusion.py
index 7327522..c88765c 100755
--- a/minidiffusion.py
+++ b/minidiffusion.py
@@ -289,6 +289,9 @@ model.eval()
 if train_input.dim() == 2 and train_input.size(1) == 1:
 
     fig = plt.figure()
+    fig.set_figheight(5)
+    fig.set_figwidth(8)
+
     ax = fig.add_subplot(1, 1, 1)
 
     x = generate((10000, 1), T, alpha, alpha_bar, sigma,
@@ -315,7 +318,7 @@ if train_input.dim() == 2 and train_input.size(1) == 1:
     fig.savefig(filename, bbox_inches='tight')
 
     if not args.no_window and hasattr(plt.get_current_fig_manager(), 'window'):
-        plt.get_current_fig_manager().window.setGeometry(2, 2, 2048, 768)
+        plt.get_current_fig_manager().window.setGeometry(2, 2, 1024, 768)
         plt.show()
 
 ########################################
@@ -323,6 +326,9 @@ if train_input.dim() == 2 and train_input.size(1) == 1:
 elif train_input.dim() == 2 and train_input.size(1) == 2:
 
     fig = plt.figure()
+    fig.set_figheight(6)
+    fig.set_figwidth(6)
+
     ax = fig.add_subplot(1, 1, 1)
 
     x = generate((1000, 2), T, alpha, alpha_bar, sigma,