From 52a502df8b6f08031a509ac5c55d2b60d92361f7 Mon Sep 17 00:00:00 2001 From: Macrobull Date: Thu, 28 Mar 2019 21:13:57 +0800 Subject: [PATCH] bugfix --- onnx2fluid/onnx2fluid/validation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onnx2fluid/onnx2fluid/validation.py b/onnx2fluid/onnx2fluid/validation.py index fd6b5c1..a487850 100644 --- a/onnx2fluid/onnx2fluid/validation.py +++ b/onnx2fluid/onnx2fluid/validation.py @@ -95,8 +95,8 @@ def validate(fluid_model_filename, output_data = test_data['outputs'].tolist() else: test_data = np.load(golden_data_filename).tolist() - input_data = input_data['inputs'] - output_data = output_data['outputs'] + input_data = test_data['inputs'] + output_data = test_data['outputs'] input_data = _flatten_dict(input_data) output_data = _flatten_dict(output_data) logger.info('found %d I/O golden data, starting test ...', len(test_data)) -- GitLab