Untitled

 avatar
unknown
plain_text
4 months ago
256 B
3
Indexable

# Forward measurement model (Ax + n)
y = forward(ref_img, task, mask_type)

y_n = noiser(y)


# Sampling
x_start = torch.randn(ref_img.shape, device=device).requires_grad_()
sample = sdedit(model, x_start, measurement=y_n, record=True, save_root=out_path)
Editor is loading...
Leave a Comment