Register backward hook for the whole optim_dict to enable working at multi schedule pp

Register backward hook for the whole optim_dict to enable working at multi schedule pp #1777