From ce2da55e34be7013c775b96bd82d8cb2d3d64d6a Mon Sep 17 00:00:00 2001 From: Mergen Nachin Date: Tue, 8 Apr 2025 16:10:01 -0400 Subject: [PATCH] Getting Started, compare against reference eager model for verification in pybind --- docs/source/getting-started.md | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/docs/source/getting-started.md b/docs/source/getting-started.md index fbca80cf23b..a3eb056fb32 100644 --- a/docs/source/getting-started.md +++ b/docs/source/getting-started.md @@ -86,7 +86,17 @@ runtime = Runtime.get() input_tensor: torch.Tensor = torch.randn(1, 3, 224, 224) program = runtime.load_program("model.pte") method = program.load_method("forward") -outputs: List[torch.Tensor] = method.execute([input_tensor]) +output: List[torch.Tensor] = method.execute([input_tensor]) +print("Run succesfully via executorch") + +from torchvision.models.mobilenetv2 import MobileNet_V2_Weights +import torchvision.models as models + +eager_reference_model = models.mobilenetv2.mobilenet_v2(weights=MobileNet_V2_Weights.DEFAULT).eval() +eager_reference_output = eager_reference_model(input_tensor) + +print("Comparing against original PyTorch module") +print(torch.allclose(output[0], eager_reference_output, rtol=1e-3, atol=1e-5)) ```