提交 3ff10601 编写于 作者: Y Yang Zhang 提交者: qingqing01

Fix nolocal naming (#24)

* Fix nolocal naming
* Fix `cb_resnet` as well
上级 a992fef4
......@@ -28,7 +28,7 @@ from ppdet.core.workspace import register, serializable
from numbers import Integral
from .name_adapter import NameAdapter
from .nonlocal import add_space_nonlocal
from .nonlocal_helper import add_space_nonlocal
__all__ = ['CBResNet']
......@@ -423,4 +423,3 @@ class CBResNet(object):
return OrderedDict([('res{}_sum'.format(self.feature_maps[idx]), feat)
for idx, feat in enumerate(res_endpoints)])
......@@ -27,7 +27,7 @@ from paddle.fluid.initializer import Constant
from ppdet.core.workspace import register, serializable
from numbers import Integral
from .nonlocal import add_space_nonlocal
from .nonlocal_helper import add_space_nonlocal
from .name_adapter import NameAdapter
__all__ = ['ResNet', 'ResNetC5']
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册