projects
/
pytorch.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
|
inline
| side by side (parent:
e8500a3
)
Update.
author
Francois Fleuret
<francois@fleuret.org>
Wed, 17 Aug 2022 06:14:36 +0000
(08:14 +0200)
committer
Francois Fleuret
<francois@fleuret.org>
Wed, 17 Aug 2022 06:14:36 +0000
(08:14 +0200)
minidiffusion.py
patch
|
blob
|
history
diff --git
a/minidiffusion.py
b/minidiffusion.py
index
7327522
..
c88765c
100755
(executable)
--- 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()
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,
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'):
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()
########################################
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()
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,
ax = fig.add_subplot(1, 1, 1)
x = generate((1000, 2), T, alpha, alpha_bar, sigma,