Skip to content

Commit 92c9a24

Browse files
esythanroot
authored andcommitted
change code style
1 parent 240c622 commit 92c9a24

File tree

4 files changed

+14
-21
lines changed

4 files changed

+14
-21
lines changed

models/rank/slot_dnn/inmemorydataset_reader.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -67,9 +67,10 @@ def line_process(self, line):
6767
if slot not in self.slots:
6868
continue
6969
feasign = int(slot_feasign[0])
70-
if feasign not in fea_dict:
71-
fea_dict[feasign] = str(len(fea_dict))
72-
output[self.slot2index[slot]][1].append(fea_dict[feasign])
70+
# if feasign not in fea_dict:
71+
# fea_dict[feasign] = str(len(fea_dict))
72+
# output[self.slot2index[slot]][1].append(fea_dict[feasign])
73+
output[self.slot2index[slot]][1].append(feasign)
7374
self.visit[slot] = True
7475
for i in self.visit:
7576
slot = i

models/rank/slot_dnn/net.py

Lines changed: 5 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -67,20 +67,11 @@ def forward(self, slot_inputs):
6767
embs = []
6868
self.inference_feed_vars = []
6969
for s_input in slot_inputs:
70-
if self.sync_mode == "gpubox":
71-
emb = paddle.static.nn.sparse_embedding(
72-
#emb = paddle.fluid.contrib.sparse_embedding(
73-
input=s_input,
74-
size=[self.dict_dim, self.emb_dim],
75-
param_attr=paddle.ParamAttr(name="embedding"))
76-
else:
77-
emb = paddle.fluid.layers.embedding(
78-
input=s_input,
79-
size=[self.dict_dim, self.emb_dim],
80-
is_sparse=True,
81-
is_distributed=True,
82-
param_attr=paddle.ParamAttr(name="embedding"))
83-
#emb = self.embedding(s_input)
70+
emb = paddle.static.nn.sparse_embedding(
71+
#emb = paddle.fluid.contrib.sparse_embedding(
72+
input=s_input,
73+
size=[self.dict_dim, self.emb_dim],
74+
param_attr=paddle.ParamAttr(name="embedding"))
8475
self.inference_feed_vars.append(emb)
8576

8677
bow = paddle.fluid.layers.sequence_pool(input=emb, pool_type='sum')

models/rank/slot_dnn/queuedataset_reader.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,9 +66,10 @@ def line_process(self, line):
6666
if slot not in self.slots:
6767
continue
6868
feasign = int(slot_feasign[0])
69-
if feasign not in fea_dict:
70-
fea_dict[feasign] = len(fea_dict)
71-
output[self.slot2index[slot]][1].append(fea_dict[feasign])
69+
# if feasign not in fea_dict:
70+
# fea_dict[feasign] = len(fea_dict)
71+
# output[self.slot2index[slot]][1].append(fea_dict[feasign])
72+
output[self.slot2index[slot]][1].append(feasign)
7273
self.visit[slot] = True
7374
for i in self.visit:
7475
slot = i

tools/utils/static_ps/flow_helper.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ def save_inference_model(exe,
139139
"""
140140
day = str(day)
141141
pass_id = str(pass_id)
142-
suffix_name = "/%s/%s/inference_model/" % (day, pass_id)
142+
suffix_name = "/%s/inference_model_%s/" % (day, pass_id)
143143
model_path = output_path + suffix_name
144144
logger.info("going to save_inference_model %s" % model_path)
145145
fleet.save_inference_model(

0 commit comments

Comments
 (0)