Skip to content

Commit 39bc69a

Browse files
committed
fix save_model
1 parent 161ef56 commit 39bc69a

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tools/static_trainer.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,12 @@ def main(args):
132132
else:
133133
logger.info("reader type wrong")
134134

135+
save_static_model(
136+
paddle.static.default_main_program(),
137+
model_save_path,
138+
epoch_id,
139+
prefix='rec_static')
140+
135141
if use_inference:
136142
feed_var_names = config.get("runner.save_inference_feed_varnames",
137143
[])
@@ -164,12 +170,6 @@ def main(args):
164170

165171
save_inference_model(model_save_path, epoch_id, feedvars,
166172
fetchvars, exe)
167-
else:
168-
save_static_model(
169-
paddle.static.default_main_program(),
170-
model_save_path,
171-
epoch_id,
172-
prefix='rec_static')
173173

174174

175175
def dataset_train(epoch_id, dataset, fetch_vars, exe, config):

0 commit comments

Comments
 (0)