Skip to content

Commit fe8d103

Browse files
authored
update save model (#6697)
1 parent ab30f23 commit fe8d103

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ppocr/utils/save_load.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -177,9 +177,9 @@ def save_model(model,
177177
model.backbone.model.save_pretrained(model_prefix)
178178
metric_prefix = os.path.join(model_prefix, 'metric')
179179
# save metric and config
180+
with open(metric_prefix + '.states', 'wb') as f:
181+
pickle.dump(kwargs, f, protocol=2)
180182
if is_best:
181-
with open(metric_prefix + '.states', 'wb') as f:
182-
pickle.dump(kwargs, f, protocol=2)
183183
logger.info('save best model is to {}'.format(model_prefix))
184184
else:
185185
logger.info("save model in {}".format(model_prefix))

0 commit comments

Comments
 (0)