- x = generate((128,) + train_input.size()[1:], T, alpha, alpha_bar, sigma,
- model, train_mean, train_std)
-
- x = torchvision.utils.make_grid(x.clamp(min = 0, max = 255),
- nrow = 16, padding = 1, pad_value = 64)
- x = F.pad(x, pad = (2, 2, 2, 2), value = 64)[None]
+ x = torchvision.utils.make_grid(
+ x.clamp(min=0, max=255), nrow=16, padding=1, pad_value=64
+ )
+ x = F.pad(x, pad=(2, 2, 2, 2), value=64)[None]