(self, sigma)
| 81 | return (sigma ** 2 + self.sigma_data ** 2) / (sigma * self.sigma_data) ** 2 |
| 82 | |
| 83 | def get_scalings(self, sigma): |
| 84 | c_skip = self.sigma_data ** 2 / (sigma ** 2 + self.sigma_data ** 2) |
| 85 | c_out = sigma * self.sigma_data / (sigma ** 2 + self.sigma_data ** 2) ** 0.5 |
| 86 | c_in = 1 / (sigma ** 2 + self.sigma_data ** 2) ** 0.5 |
| 87 | return c_skip, c_out, c_in |
| 88 | |
| 89 | def loss(self, input, noise, sigma, **kwargs): |
| 90 | # print("Denoiser") |