Use enable_gradient_checkpointing() for flux#244
Open
jhj0517 wants to merge 1 commit intoostris:mainfrom
Open
Use `enable_gradient_checkpointing()` for flux#244jhj0517 wants to merge 1 commit intoostris:mainfrom
jhj0517 wants to merge 1 commit intoostris:mainfrom