Skip to content

Commit de88a68

Browse files
authored
Merge pull request #402 from frankwhzhang/fix_esmm_para_0324
fix esmm model parameters bug
2 parents 6df90ff + 6ecee14 commit de88a68

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

models/multitask/esmm/net.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,8 @@ def __init__(self, sparse_feature_number, sparse_feature_dim, num_field,
6868
weight_attr=paddle.ParamAttr(
6969
initializer=paddle.nn.initializer.Normal(
7070
std=1.0 / math.sqrt(cvr_sizes[i]))))
71-
self.add_sublayer('linear_%d' % i, linear)
71+
self.add_sublayer('linear_%d' % (len(ctr_layer_sizes) + 1 + i),
72+
linear)
7273
self._cvr_mlp_layers.append(linear)
7374
if acts[i] == 'relu':
7475
act = paddle.nn.ReLU()

0 commit comments

Comments
 (0)