Skip to content

Commit 260f6db

Browse files
committed
fix fat
1 parent e6e00ac commit 260f6db

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

models/rank/fat_deepffm/net.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -193,6 +193,7 @@ def __init__(self,
193193
if acts[i] == 'relu':
194194
act = paddle.nn.ReLU()
195195
self.add_sublayer('act_%d' % i, act)
196+
self._mlp_layers.append(act)
196197

197198
def forward(self, feat_embeddings):
198199
y_dnn = paddle.reshape(feat_embeddings, [feat_embeddings.shape[0], -1])

0 commit comments

Comments
 (0)