ZF/insightface
1
0
Fork 0
mirror of https://github.com/deepinsight/insightface.git synced 2026-05-17 14:26:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

new prefix for model slim

Browse Source
This commit is contained in:
nttstar
2018-10-04 20:04:03 +08:00
parent 744fa87348
commit bc1cf15c50
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
deploy/model_slim.py
View File

@@ -27,5 +27,5 @@ for k,v in arg_params.iteritems():
dellist.append(k)
for d in dellist:
del arg_params[d]
mx.model.save_checkpoint(prefix, 0, sym, arg_params, aux_params)
mx.model.save_checkpoint(prefix+"s", 0, sym, arg_params, aux_params)