set_switch_weight_model function

Set switching weight model