Browse Source

feat(mge): do not export F.nn.sync_batch_norm

GitOrigin-RevId: aa05e3aac4
release-1.1
Megvii Engine Team 4 years ago
parent
commit
7589bce454
2 changed files with 1 additions and 2 deletions
  1. +0
    -1
      imperative/python/megengine/functional/nn.py
  2. +1
    -1
      imperative/python/megengine/module/batchnorm.py

+ 0
- 1
imperative/python/megengine/functional/nn.py View File

@@ -57,7 +57,6 @@ __all__ = [
"softmax", "softmax",
"softplus", "softplus",
"svd", "svd",
"sync_batch_norm",
"warp_perspective", "warp_perspective",
] ]




+ 1
- 1
imperative/python/megengine/module/batchnorm.py View File

@@ -11,7 +11,7 @@ from typing import Optional
import numpy as np import numpy as np


from ..distributed.group import WORLD, Group 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 ..tensor import Parameter, Tensor
from . import init from . import init
from .module import Module from .module import Module


Loading…
Cancel
Save