Google Git
Sign in
android / platform / external / pytorch / 79c3bbceb2 / . / torch / distributed / fsdp
tree: 30f395c194fed6c36021c34e8f6694a8defda6ee [path history] [tgz]
  1. __init__.py
  2. flatten_params_wrapper.py
  3. fully_sharded_data_parallel.py
  4. utils.py
  5. wrap.py