Commits · e7602a4c47d7befd8177bc51cdb40a5794df982c · OpenDAS / fairscale

30 May, 2022 1 commit
- [minor] test split_non_tensor(<Tensor>) case (#1000) · e7602a4c
  Crutcher Dunnavant authored May 30, 2022
  
  e7602a4c
26 May, 2022 1 commit
- [minor] Remove false tuples (#994) · c4af33b6
  Crutcher Dunnavant authored May 26, 2022
  
  c4af33b6
08 Mar, 2021 1 commit

Fixed RNN support for containers (#494) · 8c405c51

Sean Naren authored Mar 08, 2021



* Fix packed sequence apply

* Update fairscale/utils/containers.py
Co-authored-by: Min Xu <24926999+min-xu-ai@users.noreply.github.com>

8c405c51

26 Feb, 2021 1 commit

[feat]: add summon_full_params context mgr (#433) · 77f92b38

Min Xu authored Feb 25, 2021

* [feat]: add summon_full_params context mgr

* fix

* fix

* addressed comments

* fixed the state_dict copy

* lint

77f92b38

10 Feb, 2021 1 commit

Add fairscale.nn.misc.checkpoint_activations (#376) · c963a72a

Myle Ott authored Feb 10, 2021



* Add fairscale.utils.containers
Co-authored-by: Min Xu <24926999+min-xu-ai@users.noreply.github.com>

* Add fairscale.nn.misc.checkpoint_activations
Co-authored-by: Sam Shleifer <sshleifer@gmail.com>
Co-authored-by: Min Xu <24926999+min-xu-ai@users.noreply.github.com>
Co-authored-by: Sam Shleifer <sshleifer@gmail.com>

c963a72a