From 289669a563bf3f97e08c3805e763451b4d29d8b9 Mon Sep 17 00:00:00 2001 From: bingyaweng Date: Tue, 25 Aug 2020 10:39:40 +0800 Subject: [PATCH] fix error of WithBNNLossCell --- .../api/python/mindspore/mindspore.nn.probability.rst | 3 +++ .../api/python/mindspore/mindspore.nn.probability.rst | 3 +++ 2 files changed, 6 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 ec19edcb..9ed8e369 100644 --- a/api/source_en/api/python/mindspore/mindspore.nn.probability.rst +++ b/api/source_en/api/python/mindspore/mindspore.nn.probability.rst @@ -13,6 +13,9 @@ mindspore.nn.probability.bnn_layers .. automodule:: mindspore.nn.probability.bnn_layers :members: + .. autoclass:: WithBNNLossCell + :members: + mindspore.nn.probability.distribution ------------------------------------- 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 ec19edcb..729903df 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 @@ -13,6 +13,9 @@ mindspore.nn.probability.bnn_layers .. automodule:: mindspore.nn.probability.bnn_layers :members: + .. autoclass:: WithBNNLossCell + :members: + mindspore.nn.probability.distribution ------------------------------------- -- GitLab