hub / github.com/VincentHancoder/SSGD / ort_validate

Function ort_validate

tests/test_onnx/utils.py:32–85 · view source on GitHub ↗

Validate the output of the onnxruntime backend is the same as the output generated by torch. Args: model (nn.Module | function): the function of model or model to be verified. feats (tuple(list(torch.Tensor)) | list(torch.Tensor) | torch.Tensor): the

(model, feats, onnx_io='tmp.onnx')

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 15

test_retina_head_forward_singleFunction · 0.85

test_retina_head_forwardFunction · 0.85

test_retinanet_head_onnx_exportFunction · 0.85

test_yolov3_head_forwardFunction · 0.85

test_yolov3_head_onnx_exportFunction · 0.85

test_fcos_head_forward_singleFunction · 0.85

test_fcos_head_forwardFunction · 0.85

test_fcos_head_onnx_exportFunction · 0.85

test_fsaf_head_forward_singleFunction · 0.85

test_fsaf_head_forwardFunction · 0.85

test_fsaf_head_onnx_exportFunction · 0.85

test_ssd_head_forwardFunction · 0.85

Calls 7

forwardMethod · 0.95

WrapFunctionClass · 0.85

get_ort_model_outputFunction · 0.85

convert_result_listFunction · 0.85

cpuMethod · 0.80

numpyMethod · 0.80

detachMethod · 0.80

Tested by 15

test_retina_head_forward_singleFunction · 0.68

test_retina_head_forwardFunction · 0.68

test_retinanet_head_onnx_exportFunction · 0.68

test_yolov3_head_forwardFunction · 0.68

test_yolov3_head_onnx_exportFunction · 0.68

test_fcos_head_forward_singleFunction · 0.68

test_fcos_head_forwardFunction · 0.68

test_fcos_head_onnx_exportFunction · 0.68

test_fsaf_head_forward_singleFunction · 0.68

test_fsaf_head_forwardFunction · 0.68

test_fsaf_head_onnx_exportFunction · 0.68

test_ssd_head_forwardFunction · 0.68