Google Git
Sign in
android / platform / external / pytorch / 0a651a231d81fc34d6b8858ee41cf7b9772c042a / . / torch / distributed / fsdp
tree: a0bf47291655e13032b9bb1722dce241480eabfe [path history] [tgz]
  1. __init__.py
  2. _optim_utils.py
  3. _utils.py
  4. flatten_params_wrapper.py
  5. fully_sharded_data_parallel.py
  6. shard_utils.py
  7. sharded_grad_scaler.py
  8. wrap.py