Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • LuGY's avatar
    [zero]fix zero ckptIO with offload (#4529) · cbac7822
    LuGY authored 
    * fix zero ckptio with offload

* fix load device

* saved tensors in ckpt should be on CPU

* fix unit test

* fix unit test

* add clear cache

* save memory for CI
    cbac7822
low_level_optim.py 25.5 KB