def update_scale():
    with tf.control_dependencies([perturb_for_adaption]):
        update_scale_expr = tf.cond((mean_kl < param_noise_threshold), (lambda : param_noise_scale.assign((param_noise_scale * 1.01))), (lambda : param_noise_scale.assign((param_noise_scale / 1.01))))
    return update_scale_expr
def update_scale():
    with tf.control_dependencies([perturb_for_adaption]):
        update_scale_expr = tf.cond((mean_kl < param_noise_threshold), (lambda : param_noise_scale.assign((param_noise_scale * 1.01))), (lambda : param_noise_scale.assign((param_noise_scale / 1.01))))
    return update_scale_expr