Why Gemfury? Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Debian packages RPM packages NuGet packages

team-dmm / torch python

Repository URL to install this package:

Version: 
2.4.0  ▾
Details    
torch / distributed / _composable / fsdp / __init__.py
Size: Mime: 
from ._fsdp_api import CPUOffloadPolicy, MixedPrecisionPolicy, OffloadPolicy
from .fully_shard import FSDPModule, fully_shard, register_fsdp_forward_method

Products

About

Resources

Contact Gemfury