提交 5dcd2de1 编写于 作者: M michaelowenliu

change BN in bachbones to SyncBN

上级 d0b456c6
......@@ -21,7 +21,8 @@ import paddle
import paddle.fluid as fluid
from paddle.fluid.param_attr import ParamAttr
from paddle.fluid.layer_helper import LayerHelper
from paddle.fluid.dygraph.nn import Conv2D, Pool2D, BatchNorm, Linear, Dropout
from paddle.fluid.dygraph.nn import Conv2D, Pool2D, Linear, Dropout
from paddle.fluid.dygraph import SyncBatchNorm as BatchNorm
import math
......
......@@ -24,7 +24,8 @@ import paddle
import paddle.fluid as fluid
from paddle.fluid.param_attr import ParamAttr
from paddle.fluid.layer_helper import LayerHelper
from paddle.fluid.dygraph.nn import Conv2D, Pool2D, BatchNorm, Linear, Dropout
from paddle.fluid.dygraph.nn import Conv2D, Pool2D, Linear, Dropout
from paddle.fluid.dygraph import SyncBatchNorm as BatchNorm
from dygraph.utils import utils
......
......@@ -2,7 +2,8 @@ import paddle
import paddle.fluid as fluid
from paddle.fluid.param_attr import ParamAttr
from paddle.fluid.layer_helper import LayerHelper
from paddle.fluid.dygraph.nn import Conv2D, Pool2D, BatchNorm, Linear, Dropout
from paddle.fluid.dygraph.nn import Conv2D, Pool2D, Linear, Dropout
from paddle.fluid.dygraph import SyncBatchNorm as BatchNorm
from dygraph.cvlibs import manager
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册