Skip to content

Commit 380c47b

Browse files
authored
Merge pull request #687 from wangzhen38/fix_xdeepfm
fix xdeepfm code style
2 parents 693e36c + 41d518a commit 380c47b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

models/rank/xdeepfm/net.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -223,6 +223,7 @@ def __init__(self, sparse_feature_dim, num_field, layer_sizes_dnn):
223223
if acts[i] == 'relu':
224224
act = paddle.nn.ReLU()
225225
self.add_sublayer('act_%d' % i, act)
226+
self._mlp_layers.append(act)
226227

227228
def forward(self, feat_embeddings):
228229
y_dnn = paddle.reshape(feat_embeddings,

0 commit comments

Comments
 (0)