From b19c2b7ddf3e4db73d422c3c7e6c4371f9d6e657 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Wed, 17 Aug 2022 08:14:36 +0200 Subject: [PATCH] Update. --- minidiffusion.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) 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, -- 2.20.1