package info
(click to toggle)
Folder: fsdp
![]() |
.. (parent) | |||
![]() |
- | rw-r--r-- | 329 | __init__.py |
![]() |
- | rw-r--r-- | 3,043 | _fsdp_extensions.py |
![]() |
- | rw-r--r-- | 53,007 | _optim_utils.py |
![]() |
- | rw-r--r-- | 9,373 | _shard_utils.py |
![]() |
- | rw-r--r-- | 10,161 | _symbolic_trace.py |
![]() |
- | rw-r--r-- | 4,991 | _utils.py |
![]() |
- | rw-r--r-- | 47,402 | flat_param.py |
![]() |
- | rw-r--r-- | 6,486 | flatten_params_wrapper.py |
![]() |
- | rw-r--r-- | 211,953 | fully_sharded_data_parallel.py |
![]() |
- | rw-r--r-- | 16,497 | sharded_grad_scaler.py |
![]() |
- | rw-r--r-- | 4,636 | utils.py |
![]() |
- | rw-r--r-- | 18,185 | wrap.py |