From 2a1e0624ab75de661f92d23bed665bfbf811e021 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Thu, 30 Apr 2020 11:21:16 +0800 Subject: [PATCH] docs(mge/functional): fix doctest GitOrigin-RevId: 45b6e1147ec72cecdd7c25de2074b498cd8abc70 --- python_module/megengine/functional/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python_module/megengine/functional/utils.py b/python_module/megengine/functional/utils.py index 41013aed..0ad969ca 100644 --- a/python_module/megengine/functional/utils.py +++ b/python_module/megengine/functional/utils.py @@ -26,7 +26,8 @@ def _decide_comp_node_and_comp_graph(*args: mgb.SymbolVar): def accuracy( logits: Tensor, target: Tensor, topk: Union[int, Iterable[int]] = 1 ) -> Union[Tensor, Iterable[Tensor]]: - r"""Calculate the classification accuracy given predicted logits and ground-truth labels. + r""" + Calculate the classification accuracy given predicted logits and ground-truth labels. :param logits: Model predictions of shape [batch_size, num_classes], representing the probability (likelyhood) of each class.