From 7589bce454dfe33d5ebf90d08f98140a096f4849 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Sat, 10 Oct 2020 19:59:26 +0800 Subject: [PATCH] feat(mge): do not export F.nn.sync_batch_norm GitOrigin-RevId: aa05e3aac4529413d9dd71a6a54d5d97e2100740 --- imperative/python/megengine/functional/nn.py | 1 - imperative/python/megengine/module/batchnorm.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py index 3da95a79..11c95972 100644 --- a/imperative/python/megengine/functional/nn.py +++ b/imperative/python/megengine/functional/nn.py @@ -57,7 +57,6 @@ __all__ = [ "softmax", "softplus", "svd", - "sync_batch_norm", "warp_perspective", ] diff --git a/imperative/python/megengine/module/batchnorm.py b/imperative/python/megengine/module/batchnorm.py index 0bc849f3..5906a459 100644 --- a/imperative/python/megengine/module/batchnorm.py +++ b/imperative/python/megengine/module/batchnorm.py @@ -11,7 +11,7 @@ from typing import Optional import numpy as np from ..distributed.group import WORLD, Group -from ..functional import batch_norm2d, sync_batch_norm +from ..functional.nn import batch_norm2d, sync_batch_norm from ..tensor import Parameter, Tensor from . import init from .module import Module