Skip to content

Commit dc11cfa

Browse files
committed
fix a test
1 parent 358159a commit dc11cfa

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

onnx_diagnostic/torch_models/validate.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -915,13 +915,13 @@ def node_iter(proto):
915915
def _validate_do_run_model(
916916
data, summary, key, tag, expected_tag, verbose, repeat, warmup, quiet
917917
):
918-
if verbose:
919-
print(f"[validate_model] -- run the model inputs={key!r}...")
920-
print(f"[validate_model] {key}={string_type(data[key], with_shape=True)}")
921918
if key not in data:
922919
if verbose:
923920
print(f"[validate_model] input; {key!r} not defined, skip.")
924921
return
922+
if verbose:
923+
print(f"[validate_model] -- run the model inputs={key!r}...")
924+
print(f"[validate_model] {key}={string_type(data[key], with_shape=True)}")
925925
# We make a copy of the input just in case the model modifies them inplace
926926
hash_inputs = string_type(data[key], with_shape=True)
927927
inputs = torch_deepcopy(data[key])

0 commit comments

Comments
 (0)