提交 c8b3439c 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!770 FIxi the API params in web.

Merge pull request !770 from zhangyi/fix_api_param
...@@ -12,7 +12,14 @@ mindspore.nn.probability.bnn_layers ...@@ -12,7 +12,14 @@ mindspore.nn.probability.bnn_layers
.. automodule:: mindspore.nn.probability.bnn_layers .. automodule:: mindspore.nn.probability.bnn_layers
:members: :members:
:exclude-members: ConvReparam , DenseReparam
.. autoclass:: ConvReparam(in_channels, out_channels, kernel_size,stride=1, pad_mode='same', padding=0, dilation=1, group=1, has_bias=False, weight_prior_fn=NormalPrior, weight_posterior_fn=<lambda name, shape: NormalPosterior(name=name, shape=shape)>, bias_prior_fn=NormalPrior, bias_posterior_fn=<lambda name, shape: NormalPosterior(name=name, shape=shape)>)
:members:
.. autoclass:: DenseReparam(in_channels, out_channels, activation=None, has_bias=True, weight_prior_fn=NormalPrior, weight_posterior_fn=<lambda name, shape: NormalPosterior(name=name, shape=shape)>, bias_prior_fn=NormalPrior, bias_posterior_fn=<lambda name, shape: NormalPosterior(name=name, shape=shape)>)
:members:
.. autoclass:: WithBNNLossCell .. autoclass:: WithBNNLossCell
:members: :members:
......
...@@ -12,7 +12,14 @@ mindspore.nn.probability.bnn_layers ...@@ -12,7 +12,14 @@ mindspore.nn.probability.bnn_layers
.. automodule:: mindspore.nn.probability.bnn_layers .. automodule:: mindspore.nn.probability.bnn_layers
:members: :members:
:exclude-members: ConvReparam , DenseReparam
.. autoclass:: ConvReparam(in_channels, out_channels, kernel_size,stride=1, pad_mode='same', padding=0, dilation=1, group=1, has_bias=False, weight_prior_fn=NormalPrior, weight_posterior_fn=<lambda name, shape: NormalPosterior(name=name, shape=shape)>, bias_prior_fn=NormalPrior, bias_posterior_fn=<lambda name, shape: NormalPosterior(name=name, shape=shape)>)
:members:
.. autoclass:: DenseReparam(in_channels, out_channels, activation=None, has_bias=True, weight_prior_fn=NormalPrior, weight_posterior_fn=<lambda name, shape: NormalPosterior(name=name, shape=shape)>, bias_prior_fn=NormalPrior, bias_posterior_fn=<lambda name, shape: NormalPosterior(name=name, shape=shape)>)
:members:
.. autoclass:: WithBNNLossCell .. autoclass:: WithBNNLossCell
:members: :members:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册