Google Git
Sign in
android / platform / external / pytorch / e8742f15cf / . / torch / distributed / _fsdp
tree: 135fd75a97ef77391e18c1ad8ce340f4e3b1e609 [path history] [tgz]
  1. __init__.py
  2. flatten_params_wrapper.py
  3. fully_sharded_data_parallel.py
  4. utils.py