Commit de33175f by BoxuanXu

fix converter try except bug

1 parent a72aa6a4
Showing with 8 additions and 2 deletions
...@@ -401,6 +401,7 @@ def test(): ...@@ -401,6 +401,7 @@ def test():
logging.info(load_graph(sym)) logging.info(load_graph(sym))
def load_checkpoint(params, network_struct): def load_checkpoint(params, network_struct):
try:
symbol = mx.sym.load(network_struct) symbol = mx.sym.load(network_struct)
save_dict = mx.nd.load(params) save_dict = mx.nd.load(params)
arg_params = {} arg_params = {}
...@@ -412,6 +413,9 @@ def load_checkpoint(params, network_struct): ...@@ -412,6 +413,9 @@ def load_checkpoint(params, network_struct):
if tp == 'aux': if tp == 'aux':
aux_params[name] = v aux_params[name] = v
return (symbol, arg_params, aux_params) return (symbol, arg_params, aux_params)
except Exception, e:
logging.info('model load failed!!!')
return (None, None, None)
#function created by xuboxuan@20170807 #function created by xuboxuan@20170807
#if __name__ == '__main__': #if __name__ == '__main__':
...@@ -420,12 +424,14 @@ def Run_Converter(model_param,model_json,seetanet_model): ...@@ -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_param',type=str,default = None)
#parser.add_argument('--model_json',type=str,default = None) #parser.add_argument('--model_json',type=str,default = None)
#args = parser.parse_args() #args = parser.parse_args()
#model_param = "wKgB6Fmo2w2ASRqpBky2APcM8zs.params" #model_param = "wKgB7Vm2iBaALVPdENbR4G1D6sc.params"
#model_json = "wKgB6Vmo2w2AXrJbAAGjO2NrZLE75.json" #model_json = "wKgB6lm2iBaAcq2PAAANvOLMPRI82.json"
#seetanet_model = "model_test" #seetanet_model = "model_test"
try: try:
sym, arg_params, aux_params = \ sym, arg_params, aux_params = \
load_checkpoint(model_param, model_json) 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) graph = load_graph(model_json)
converter = Converter(graph, arg_params, aux_params) converter = Converter(graph, arg_params, aux_params)
logging.info('start to convert model parameters') logging.info('start to convert model parameters')
......
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!