Skip to content

Commit fd73f2b

Browse files
committed
dist lookup table only support sgd while other dense table support all optimizer
1 parent 0964de1 commit fd73f2b

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

python/paddle/fluid/transpiler/distribute_transpiler.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -896,8 +896,6 @@ def _create_table_optimize_block(self, pserver_index, pserver_program,
896896
self.table_name
897897
][0]
898898
table_opt_block = pserver_program.create_block(pre_block_idx)
899-
# only support sgd now
900-
assert table_opt_op.type == "sgd"
901899

902900
if self.sync_mode:
903901
# create grad vars in pserver program
@@ -937,11 +935,8 @@ def _create_table_optimize_block(self, pserver_index, pserver_program,
937935
"LearningRate": [lr_var]
938936
}
939937
outputs = {"ParamOut": [param_var]}
940-
table_opt_block.append_op(
941-
type=table_opt_op.type,
942-
inputs=inputs,
943-
outputs=outputs,
944-
attrs=table_opt_op.attrs)
938+
# only support sgd now
939+
table_opt_block.append_op(type="sgd", inputs=inputs, outputs=outputs)
945940

946941
# add table parameter gradient and it's block id to grad_to_block_id
947942
grad_to_block_id.append(grad_var.name + ":" + str(table_opt_block.idx))

0 commit comments

Comments
 (0)