set_timestep_idx 函数功能设置当前迭代时间步。 函数原型def set_timestep_idx(cls, cur_timestep: int) -> None: 参数说明 参数名 输入/输出 类型 说明 cur_timestep 输入 int 当前迭代的时间步,传入的时间步范围和量化权重里的范围不一致会报错。 父主题: class TimestepManager