Fix yolov4 failed in tuning with KeyError
@@ -136,7 +136,8 @@ class ONNXRTAugment:
tensors_to_dump.remove(node.output[1])
elif node.op_type == 'Resize':
tensors_to_dump.remove(node.input[1])
- if len(node.input) == 4:
+ if len(node.input) == 4: # pragma: no cover
+ if node.input[2] in tensors_to_dump:
tensors_to_dump.remove(node.input[2])
elif weight_only:
for input in node.input:
