提交 7589bce4 编写于 作者: M Megvii Engine Team

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

GitOrigin-RevId: aa05e3aac4529413d9dd71a6a54d5d97e2100740
上级 d4e2d272
......@@ -57,7 +57,6 @@ __all__ = [
"softmax",
"softplus",
"svd",
"sync_batch_norm",
"warp_perspective",
]
......
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册