Fix #647 gradient_update_fn compatible with AdamW#648
Open
ShanningZhuang wants to merge 1 commit intogoogle:mainfrom
Open
Fix #647 gradient_update_fn compatible with AdamW#648ShanningZhuang wants to merge 1 commit intogoogle:mainfrom
ShanningZhuang wants to merge 1 commit intogoogle:mainfrom
