From d22d1676e7487761a40c5674ec70ba8d92d1a5dc Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 9 Oct 2020 22:34:44 +0800 Subject: [PATCH] feat(mge): do not export F.nn.roi_align GitOrigin-RevId: c2ac7bab326019132640974472067b9454d5d200 --- imperative/python/megengine/functional/nn.py | 3 +-- imperative/python/test/unit/functional/test_functional.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py index 13dee313..c092f8a7 100644 --- a/imperative/python/megengine/functional/nn.py +++ b/imperative/python/megengine/functional/nn.py @@ -45,7 +45,6 @@ __all__ = [ "max_pool2d", "one_hot", "prelu", - "roi_align", "roi_pooling", "softmax", "softplus", @@ -1382,7 +1381,7 @@ def roi_align( np.random.seed(42) inp = tensor(np.random.randn(1, 1, 128, 128)) rois = tensor(np.random.random((4, 5))) - y = F.roi_align(inp, rois, (2, 2)) + y = F.nn.roi_align(inp, rois, (2, 2)) print(y.numpy()[0]) Outputs: diff --git a/imperative/python/test/unit/functional/test_functional.py b/imperative/python/test/unit/functional/test_functional.py index 15771e0b..992d8573 100644 --- a/imperative/python/test/unit/functional/test_functional.py +++ b/imperative/python/test/unit/functional/test_functional.py @@ -170,7 +170,7 @@ def test_roi_align(): grad = Grad().wrt(inp_feat, callback=_save_to(inp_feat)) output_shape = (7, 7) - out_feat = F.roi_align( + out_feat = F.nn.roi_align( inp_feat, rois, output_shape=output_shape,