From 3f11b4213a23374311aca1c4199efd617b089d0a Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 1 Apr 2022 12:09:42 +0800 Subject: [PATCH] docs(functional): replace megengine function testcode with doctest format GitOrigin-RevId: f91d5f5afdd653c48179ec420b87d9d05101406b --- imperative/python/megengine/functional/nn.py | 2 -- imperative/python/megengine/module/lrn.py | 2 -- 2 files changed, 4 deletions(-) diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py index 56de2ea2..834d16d9 100644 --- a/imperative/python/megengine/functional/nn.py +++ b/imperative/python/megengine/functional/nn.py @@ -1850,8 +1850,6 @@ def local_response_norm( ... [19.420258, 20.331186, 21.233786, 22.127764, 23.012836 ]]]]) >>> out = F.local_response_norm(inp, kernel_size=3, k=1.0, alpha=1e-4, beta=0.75) >>> np.testing.assert_allclose(GT, out.numpy(), rtol=1e-6, atol=1e-6) - >>> print('pass') - pass """ op = builtin.LRN(n=kernel_size, k=k, alpha=alpha, beta=beta,) (output,) = apply(op, inp) diff --git a/imperative/python/megengine/module/lrn.py b/imperative/python/megengine/module/lrn.py index f509b3ff..b8b25151 100644 --- a/imperative/python/megengine/module/lrn.py +++ b/imperative/python/megengine/module/lrn.py @@ -33,8 +33,6 @@ class LocalResponseNorm(Module): >>> op = M.LocalResponseNorm(kernel_size=3, k=1.0, alpha=1e-4, beta=0.75) >>> out = op(inp) >>> np.testing.assert_allclose(GT, out.numpy(), rtol=1e-6, atol=1e-6) - >>> print('pass') - pass """ def __init__(