Skip to content

Commit ae9b357

Browse files
committed
store_path added to caffe2keras.py
1 parent 859515e commit ae9b357

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

keras/caffe/caffe2keras.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,8 @@
2727
def main(args):
2828
if(not args.store_path):
2929
store_path = args.load_path
30+
else:
31+
store_path = args.store_path
3032

3133
print("Converting model...")
3234
model = convert.caffe_to_keras(args.load_path+'/'+args.prototxt, args.load_path+'/'+args.caffemodel, debug=args.debug)

0 commit comments

Comments
 (0)