skip op py file

revert-15207-remove_op_handle_lock_and_fix_var
heqiaozhi 6 years ago
parent 016a068775
commit 60d71a9e29

@ -66,5 +66,6 @@ class DownpourSGD(object):
# Todo(guru4elephant): figure out how to support more sparse parameters
# currently only support lookup_table
worker_skipped_ops = ["lookup_table", "lookup_table_grad"]
ps_param.trainer_param.skip_op.extend(worker_skipped_ops)
ps_param_str = text_format.MessageToString(ps_param)
return [ps_param, worker_skipped_ops]

@ -60,7 +60,7 @@ class DownpourServer(Server):
table.accessor.dense_sgd_param.adam.mom_decay_rate = 0.99
table.accessor.dense_sgd_param.naive.learning_rate = 0.0002
fea_dim = 0
for param in param_var:
for param in filter(lambda x: x.name.find("embedding") == -1, param_var):
fea_dim += reduce(lambda x, y: x * y, param.shape, 1)
table.accessor.fea_dim = fea_dim

File diff suppressed because one or more lines are too long
Loading…
Cancel
Save