Acasă Explorează Ajutor
Înregistrare Autentificare
XMD
/
Lightstar
1
0
Bifurcare 0
Fisiere Probleme 0 Trageți solicitările 0 Wiki
Arbore: 3b57de02e3
Ramuri Etichete
Lightstar
/
venv
/
Lib
/
site-packages
/
torch
/
distributed
/
fsdp
/
__init__.py

__init__.py 938 B
Istoric Crud

1234567891011121314151617181920212223242526272829303132333435363738 
  1. from ._flat_param import FlatParameter as FlatParameter
    2. 

  2. from .fully_sharded_data_parallel import (
    3. 

  3.     BackwardPrefetch,
    4. 

  4.     CPUOffload,
    5. 

  5.     FullOptimStateDictConfig,
    6. 

  6.     FullStateDictConfig,
    7. 

  7.     FullyShardedDataParallel,
    8. 

  8.     LocalOptimStateDictConfig,
    9. 

  9.     LocalStateDictConfig,
    10. 

  10.     MixedPrecision,
    11. 

  11.     OptimStateDictConfig,
    12. 

  12.     OptimStateKeyType,
    13. 

  13.     ShardedOptimStateDictConfig,
    14. 

  14.     ShardedStateDictConfig,
    15. 

  15.     ShardingStrategy,
    16. 

  16.     StateDictConfig,
    17. 

  17.     StateDictSettings,
    18. 

  18.     StateDictType,
    19. 

  19. )
    20. 

  20. 

  21. __all__ = [
    22. 

  22.     "BackwardPrefetch",
    23. 

  23.     "CPUOffload",
    24. 

  24.     "FullOptimStateDictConfig",
    25. 

  25.     "FullStateDictConfig",
    26. 

  26.     "FullyShardedDataParallel",
    27. 

  27.     "LocalOptimStateDictConfig",
    28. 

  28.     "LocalStateDictConfig",
    29. 

  29.     "MixedPrecision",
    30. 

  30.     "OptimStateDictConfig",
    31. 

  31.     "OptimStateKeyType",
    32. 

  32.     "ShardedOptimStateDictConfig",
    33. 

  33.     "ShardedStateDictConfig",
    34. 

  34.     "ShardingStrategy",
    35. 

  35.     "StateDictConfig",
    36. 

  36.     "StateDictSettings",
    37. 

  37.     "StateDictType",
    38. 

  38. ]
    39.