Skip to content

Commit 94cfa75

Browse files
esythanroot
authored andcommitted
offline infer
1 parent 4c1feef commit 94cfa75

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tools/static_ps_offline_infer.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -143,11 +143,9 @@ def dataset_offline_infer(self, cur_dataset):
143143
print_step = int(config.get("runner.print_interval"))
144144
dump_fields_path = self.config.get("runner.dump_fields_path")
145145
dump_fields = [var.name for var in self.all_vars]
146-
dump_param = ["linear_1.w_0", "linear_3.w_0", "linear_5.w_0", "linear_1.b_0", "linear_3.b_0", "linear_5.b_0"]
147146
set_dump_config(paddle.static.default_main_program(), {
148147
"dump_fields_path": dump_fields_path,
149-
"dump_fields": dump_fields,
150-
"dump_param": dump_param
148+
"dump_fields": dump_fields
151149
})
152150
print(paddle.static.default_main_program()._fleet_opt)
153151

0 commit comments

Comments
 (0)