返回顶部 def register 函数功能用来注册时间步对应的策略。 函数原型def register(self, step_range, strategy): 参数说明 参数名 输入/输出 类型 说明 step_range 输入 range、list或int 对应strategy的时间步,表示一定范围的时间步或某个时间步采用的策略,如果未设置所有时间步,则采用默认策略。 strategy 输入 static或dynamic 对应输入时间步的策略,时间步的默认策略是dynamic。 父主题: class TimestepPolicyConfig