return result
samplers = {
- 'gaussian_mixture': sample_gaussian_mixture,
- 'ramp': sample_ramp,
- 'two_discs': sample_two_discs,
- 'disc_grid': sample_disc_grid,
- 'spiral': sample_spiral,
- 'mnist': sample_mnist,
+ f.__name__.removeprefix('sample_') : f for f in [
+ sample_gaussian_mixture,
+ sample_ramp,
+ sample_two_discs,
+ sample_disc_grid,
+ sample_spiral,
+ sample_mnist,
+ ]
}
######################################################################
ax.legend(frameon = False, loc = 2)
- filename = f'diffusion_{args.data}.pdf'
+ filename = f'minidiffusion_{args.data}.pdf'
print(f'saving {filename}')
fig.savefig(filename, bbox_inches='tight')
ax.legend(frameon = False, loc = 2)
- filename = f'diffusion_{args.data}.pdf'
+ filename = f'minidiffusion_{args.data}.pdf'
print(f'saving {filename}')
fig.savefig(filename, bbox_inches='tight')
result = 1 - torch.cat((t, x), 2) / 255
- filename = f'diffusion_{args.data}.png'
+ filename = f'minidiffusion_{args.data}.png'
print(f'saving {filename}')
torchvision.utils.save_image(result, filename)