p19647583
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix performance of bert

This commit is contained in:
chenhaozhe 2020-05-28 10:30:56 +08:00
parent 63479f8e7c
commit 04bc2a938e
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/pre_activate/common/helper.cc
View File

@ -686,7 +686,7 @@ bool IsSameNode(const EquivPtr &equiv1, const EquivPtr &equiv2, const VarPtr &va
MS_EXCEPTION_IF_NULL(equiv1_node);
auto equiv2_node = GetAnfNodeByVar(equiv2, var_node);
MS_EXCEPTION_IF_NULL(equiv2_node);
return equiv1_node == equiv2_node;
return *equiv1_node == *equiv2_node;
}
AnfNodePtr GetAnfNodeByVar(const EquivPtr &equiv, const VarPtr &var_node) {

2
mindspore/nn/optim/lamb.py
View File

@ -180,7 +180,7 @@ class Lamb(Optimizer):
beta2=0.999,
eps=1e-6,
weight_decay=0.0,
decay_filter=lambda x: 'LayerNorm' not in x.name and 'bias' not in x.name):
decay_filter=lambda x: 'layernorm' not in x.name.lower() and 'bias' not in x.name.lower()):
super(Lamb, self).__init__(start_learning_rate, params)
if self.is_group:

4
mindspore/ops/_grad/grad_math_ops.py
View File

@ -191,8 +191,8 @@ def get_bprop_mul(self):
mul_func = P.Mul()
def bprop(x, y, out, dout):
bc_dx = mul_func(dout, y)
bc_dy = mul_func(dout, x)
bc_dx = mul_func(y, dout)
bc_dy = mul_func(x, dout)
return binop_grad_common(x, y, bc_dx, bc_dy)
return bprop