Bagua API Documentation
v0.9.0
API Documents
bagua
bagua.torch_api
bagua.torch_api.algorithms
bagua.torch_api.checkpoint
bagua.torch_api.contrib
bagua.torch_api.data_parallel
bagua.torch_api.model_parallel
bagua.torch_api.model_parallel.moe
bagua.torch_api.bucket
bagua.torch_api.communication
bagua.torch_api.distributed
bagua.torch_api.env
bagua.torch_api.tensor
Bagua API Documentation
Bagua API Documentation
»
bagua
»
bagua.torch_api
»
bagua.torch_api.model_parallel
»
bagua.torch_api.model_parallel.moe
»
bagua.torch_api.model_parallel.moe.utils
Edit on GitHub
bagua.torch_api.model_parallel.moe.utils
¶
Module Contents
¶
bagua.torch_api.model_parallel.moe.utils.
is_moe_param
(
param
)
¶
Parameters
param
(
torch.Tensor
) –
Return type
bool
Read the Docs
v: v0.9.0
Versions
latest
stable
v0.9.0
v0.8.2
v0.8.1.post1
v0.8.1
v0.8.0
v0.7.0
v0.6.3
v0.6.2
v0.6.1
v0.6.0
0.5.0
0.4.0
Downloads
On Read the Docs
Project Home
Builds
Free document hosting provided by
Read the Docs
.