From 4a8e972c52dcbf09d9f60a0d01a5619d61cc59a1 Mon Sep 17 00:00:00 2001 From: zhangyi Date: Thu, 27 Aug 2020 20:36:18 +0800 Subject: [PATCH] fix api param's issue. --- .../api/python/mindspore/mindspore.nn.probability.rst | 7 +++++++ .../api/python/mindspore/mindspore.nn.probability.rst | 7 +++++++ 2 files changed, 14 insertions(+) diff --git a/api/source_en/api/python/mindspore/mindspore.nn.probability.rst b/api/source_en/api/python/mindspore/mindspore.nn.probability.rst index 9ed8e369..2235f574 100644 --- a/api/source_en/api/python/mindspore/mindspore.nn.probability.rst +++ b/api/source_en/api/python/mindspore/mindspore.nn.probability.rst @@ -12,7 +12,14 @@ mindspore.nn.probability.bnn_layers .. automodule:: mindspore.nn.probability.bnn_layers :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=, bias_prior_fn=NormalPrior, bias_posterior_fn=) + :members: + + .. autoclass:: DenseReparam(in_channels, out_channels, activation=None, has_bias=True, weight_prior_fn=NormalPrior, weight_posterior_fn=, bias_prior_fn=NormalPrior, bias_posterior_fn=) + :members: + .. autoclass:: WithBNNLossCell :members: diff --git a/api/source_zh_cn/api/python/mindspore/mindspore.nn.probability.rst b/api/source_zh_cn/api/python/mindspore/mindspore.nn.probability.rst index 729903df..70e8f69b 100644 --- a/api/source_zh_cn/api/python/mindspore/mindspore.nn.probability.rst +++ b/api/source_zh_cn/api/python/mindspore/mindspore.nn.probability.rst @@ -12,7 +12,14 @@ mindspore.nn.probability.bnn_layers .. automodule:: mindspore.nn.probability.bnn_layers :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=, bias_prior_fn=NormalPrior, bias_posterior_fn=) + :members: + + .. autoclass:: DenseReparam(in_channels, out_channels, activation=None, has_bias=True, weight_prior_fn=NormalPrior, weight_posterior_fn=, bias_prior_fn=NormalPrior, bias_posterior_fn=) + :members: + .. autoclass:: WithBNNLossCell :members: -- GitLab