Skip to content
Snippets Groups Projects
Commit cb9fa063 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Test] parent model bugfix in end2end

parent d238cad4
No related branches found
No related tags found
No related merge requests found
...@@ -233,7 +233,7 @@ def measure_top1_accuracy(model_chkpt, dataset, parent_chkpt=None): ...@@ -233,7 +233,7 @@ def measure_top1_accuracy(model_chkpt, dataset, parent_chkpt=None):
else: else:
parent_model = ModelWrapper(parent_chkpt) parent_model = ModelWrapper(parent_chkpt)
parent_iname = parent_model.graph.input[0].name parent_iname = parent_model.graph.input[0].name
ishape = model.get_tensor_shape(parent_iname) ishape = parent_model.get_tensor_shape(parent_iname)
ok = 0 ok = 0
nok = 0 nok = 0
n_batches = testx.shape[0] n_batches = testx.shape[0]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment