"vscode:/vscode.git/clone" did not exist on "ced53ca91904179a201777380bd8d59419d36ede"
Unverified Commit fe8d103c authored by zhoujun's avatar zhoujun Committed by GitHub
Browse files

update save model (#6697)

parent ab30f23f
......@@ -177,9 +177,9 @@ def save_model(model,
model.backbone.model.save_pretrained(model_prefix)
metric_prefix = os.path.join(model_prefix, 'metric')
# save metric and config
if is_best:
with open(metric_prefix + '.states', 'wb') as f:
pickle.dump(kwargs, f, protocol=2)
if is_best:
logger.info('save best model is to {}'.format(model_prefix))
else:
logger.info("save model in {}".format(model_prefix))
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment