Args: sigma_min (float): minimum standard deviation sigma_max (float): maximum standard deviation
(self, pi_periodic=False, cache_folder=None, sigma_min=0.01 * np.pi, sigma_max=np.pi, annealed_temp=3, mode="sde")
| 154 | @R.register('SO2VESchedule') |
| 155 | class SO2VESchedule(SO2Schedule, core.Configurable): |
| 156 | def __init__(self, pi_periodic=False, cache_folder=None, sigma_min=0.01 * np.pi, sigma_max=np.pi, annealed_temp=3, mode="sde"): |
| 157 | """ |
| 158 | Args: |
| 159 | sigma_min (float): minimum standard deviation |
| 160 | sigma_max (float): maximum standard deviation |
| 161 | """ |
| 162 | PI = 1/2 * np.pi if pi_periodic else np.pi # TODO: remove ambiguity |
| 163 | super().__init__(PI, cache_folder) |
| 164 | self.sigma_min = sigma_min |
| 165 | self.sigma_max = sigma_max |
| 166 | self.sigma_min_log = np.log(sigma_min) |
| 167 | self.sigma_max_log = np.log(sigma_max) |
| 168 | self.annealed_temp = annealed_temp |
| 169 | self.mode = mode |
| 170 | |
| 171 | def t_to_sigma(self, t): |
| 172 | """Transfer timesteps to standard deviation. |