Skip to content

Commit ce2da55

Browse files
committed
Getting Started, compare against reference eager model for verification in pybind
1 parent 8d80185 commit ce2da55

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

docs/source/getting-started.md

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,17 @@ runtime = Runtime.get()
8686
input_tensor: torch.Tensor = torch.randn(1, 3, 224, 224)
8787
program = runtime.load_program("model.pte")
8888
method = program.load_method("forward")
89-
outputs: List[torch.Tensor] = method.execute([input_tensor])
89+
output: List[torch.Tensor] = method.execute([input_tensor])
90+
print("Run succesfully via executorch")
91+
92+
from torchvision.models.mobilenetv2 import MobileNet_V2_Weights
93+
import torchvision.models as models
94+
95+
eager_reference_model = models.mobilenetv2.mobilenet_v2(weights=MobileNet_V2_Weights.DEFAULT).eval()
96+
eager_reference_output = eager_reference_model(input_tensor)
97+
98+
print("Comparing against original PyTorch module")
99+
print(torch.allclose(output[0], eager_reference_output, rtol=1e-3, atol=1e-5))
90100
```
91101

92102

0 commit comments

Comments
 (0)