Commit 1c6c061b authored by umangyadav's avatar umangyadav
Browse files

Formatting

parent 0a6bc406
......@@ -52,8 +52,14 @@ def parse_args():
parser.add_argument('--fill0',
action='store_true',
help='fill all arguments with a value of 0')
parser.add_argument('--numpy', action='append', help='fill argument with numpy saved array', type=str)
parser.add_argument('--np_path', action='append', help='Path for the saved numpy array', type=str)
parser.add_argument('--numpy',
action='append',
help='fill argument with numpy saved array',
type=str)
parser.add_argument('--np_path',
action='append',
help='Path for the saved numpy array',
type=str)
parser.add_argument('--verbose',
action='store_true',
help='show verbose information (for debugging)')
......@@ -83,7 +89,12 @@ def parse_args():
action='store_true',
default=False,
help='Turn on ort VERBOSE logging via session options')
parser.add_argument('--save-ort-res', dest="save_ort", type=str, help='Save output of ORT as numpy array at path provided by this argument')
parser.add_argument(
'--save-ort-res',
dest="save_ort",
type=str,
help=
'Save output of ORT as numpy array at path provided by this argument')
args = parser.parse_args()
return args
......@@ -191,7 +202,9 @@ def main():
if name in args.numpy:
in_path = args.np_path[args.numpy.index(name)]
if args.verbose:
print("Loading numpy array for input name : {name}, from path : {in_path}")
print(
"Loading numpy array for input name : {name}, from path : {in_path}"
)
test_input = np.load(in_path).astype(get_np_datatype(in_type))
elif not args.fill1 and not args.fill0:
test_input = np.random.rand(*(in_shape)).astype(
......@@ -223,9 +236,11 @@ def main():
try:
pred_fw = sess.run(None, ort_params)[-1]
if(args.save_ort):
if (args.save_ort):
if args.verbose:
print("saving ORT result as numpy array at location : {args.save_ort}")
print(
"saving ORT result as numpy array at location : {args.save_ort}"
)
np.save(args.save_ort, pred_fw)
except Exception as e:
if any(input_dims):
......
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