tests/nn/data_parallel/test_fsdp_freezing_weights.py · 73f73120fed1bc806d7487dcc4e999ad63e12947 · OpenDAS / fairscale · GitLab

Find file Blame History Permalink

[fix] Add an additional assert for checking if the params of a module requires_grad=True (#761) · 73f73120
anj-s authored Aug 11, 2021
```
* add additional assert for checking if the requires_grad field is set.

* fix lint errors

* add unit tests and address comments
```
73f73120

test_fsdp_freezing_weights.py 7.89 KB