代码问题

[liuxting99]

请问以下代码中的"1000",是否应该改为"noise_schedule.total_N"?因为不是所有的前向过程都是1000步
def get_model_input_time(t_continuous):
"""
Convert the continuous-time t_continuous (in [epsilon, T]) to the model input time.
For discrete-time DPMs, we convert t_continuous in [1 / N, 1] to t_input in [0, 1000 * (N - 1) / N].
For continuous-time DPMs, we just use t_continuous.
"""
if noise_schedule.schedule == 'discrete':
return (t_continuous - 1. / noise_schedule.total_N) * 1000.
else:
return t_continuous