(
t,
shift=5.0,
num_timesteps=1000,
)
| 93 | |
| 94 | |
| 95 | def timestep_transform( |
| 96 | t, |
| 97 | shift=5.0, |
| 98 | num_timesteps=1000, |
| 99 | ): |
| 100 | t = t / num_timesteps |
| 101 | # shift the timestep based on ratio |
| 102 | new_t = shift * t / (1 + (shift - 1) * t) |
| 103 | new_t = new_t * num_timesteps |
| 104 | return new_t |
| 105 | |
| 106 | |
| 107 |
no outgoing calls
no test coverage detected