Google Git
Sign in
android / platform / external / pytorch / e48ecfda9f / . / torch / distributed / fsdp
tree: 280917574c9b54e6ff18cf9e700c442fed505ef0 [path history] [tgz]
  1. __init__.py
  2. flatten_params_wrapper.py
  3. fully_sharded_data_parallel.py
  4. optim_utils.py
  5. utils.py
  6. wrap.py