Skip to content
Toggle navigation
P
Projects
G
Groups
S
Snippets
Help
BoxuanXu
/
FlaskDriverMXNet2SeetaNet
This project
Loading...
Sign in
Toggle navigation
Go to a project
Project
Repository
Issues
0
Merge Requests
0
Pipelines
Wiki
Snippets
Settings
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Commit de33175f
authored
Sep 12, 2017
by
BoxuanXu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix converter try except bug
1 parent
a72aa6a4
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
8 additions
and
2 deletions
src/converter.py
src/converter.py
View file @
de33175
...
...
@@ -401,6 +401,7 @@ def test():
logging
.
info
(
load_graph
(
sym
))
def
load_checkpoint
(
params
,
network_struct
):
try
:
symbol
=
mx
.
sym
.
load
(
network_struct
)
save_dict
=
mx
.
nd
.
load
(
params
)
arg_params
=
{}
...
...
@@ -412,6 +413,9 @@ def load_checkpoint(params, network_struct):
if
tp
==
'aux'
:
aux_params
[
name
]
=
v
return
(
symbol
,
arg_params
,
aux_params
)
except
Exception
,
e
:
logging
.
info
(
'model load failed!!!'
)
return
(
None
,
None
,
None
)
#function created by xuboxuan@20170807
#if __name__ == '__main__':
...
...
@@ -420,12 +424,14 @@ def Run_Converter(model_param,model_json,seetanet_model):
#parser.add_argument('--model_param',type=str,default = None)
#parser.add_argument('--model_json',type=str,default = None)
#args = parser.parse_args()
#model_param = "wKgB
6Fmo2w2ASRqpBky2APcM8zs
.params"
#model_json = "wKgB6
Vmo2w2AXrJbAAGjO2NrZLE75
.json"
#model_param = "wKgB
7Vm2iBaALVPdENbR4G1D6sc
.params"
#model_json = "wKgB6
lm2iBaAcq2PAAANvOLMPRI82
.json"
#seetanet_model = "model_test"
try
:
sym
,
arg_params
,
aux_params
=
\
load_checkpoint
(
model_param
,
model_json
)
if
sym
is
None
or
arg_params
is
None
or
aux_params
is
None
:
return
None
graph
=
load_graph
(
model_json
)
converter
=
Converter
(
graph
,
arg_params
,
aux_params
)
logging
.
info
(
'start to convert model parameters'
)
...
...
Write
Preview
Markdown
is supported
Attach a file
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to post a comment