Makes FSDP reshard_after_forward configurable (#822)
#1558
| Job | Run time |
|---|---|
| 18m 29s | |
| 18m 29s |
reshard_after_forward configurable (#822)
#1558
| Job | Run time |
|---|---|
| 18m 29s | |
| 18m 29s |