def piecewise_lr(step):
    return tf.compat.v1.train.piecewise_constant(step, self.step_boundaries, self.lr_values)
def piecewise_lr(step):
    return tf.compat.v1.train.piecewise_constant(step, self.step_boundaries, self.lr_values)