Skip to content

Commit 6ecee14

Browse files
committed
fix
1 parent d85651e commit 6ecee14

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)