projects
/
pytorch.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Cosmetics + added the figures.
[pytorch.git]
/
minidiffusion.py
diff --git
a/minidiffusion.py
b/minidiffusion.py
index
c88765c
..
e7be8c1
100755
(executable)
--- a/
minidiffusion.py
+++ b/
minidiffusion.py
@@
-313,7
+313,7
@@
if train_input.dim() == 2 and train_input.size(1) == 1:
ax.legend(frameon = False, loc = 2)
ax.legend(frameon = False, loc = 2)
- filename = f'diffusion_{args.data}.pdf'
+ filename = f'
mini
diffusion_{args.data}.pdf'
print(f'saving {filename}')
fig.savefig(filename, bbox_inches='tight')
print(f'saving {filename}')
fig.savefig(filename, bbox_inches='tight')
@@
-350,7
+350,7
@@
elif train_input.dim() == 2 and train_input.size(1) == 2:
ax.legend(frameon = False, loc = 2)
ax.legend(frameon = False, loc = 2)
- filename = f'diffusion_{args.data}.pdf'
+ filename = f'
mini
diffusion_{args.data}.pdf'
print(f'saving {filename}')
fig.savefig(filename, bbox_inches='tight')
print(f'saving {filename}')
fig.savefig(filename, bbox_inches='tight')
@@
-375,7
+375,7
@@
elif train_input.dim() == 4:
result = 1 - torch.cat((t, x), 2) / 255
result = 1 - torch.cat((t, x), 2) / 255
- filename = f'diffusion_{args.data}.png'
+ filename = f'
mini
diffusion_{args.data}.png'
print(f'saving {filename}')
torchvision.utils.save_image(result, filename)
print(f'saving {filename}')
torchvision.utils.save_image(result, filename)