You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
# Example of an external model for the ARM AOT Compiler
2
-
This is an example of an external Python file to be used as a module by the ```aot_arm_compiler.py``` (and the ```run.sh```) scripts in ```examples/arm``` directory.
2
+
This is an example of an external Python file to be used as a module by the `run.sh` (and the `aot_arm_compiler.py`) scripts in `examples/arm` directory. Just pass the path of the `model.py` file as `--model_name`.
3
3
4
-
```ModelUnderTest``` should be a ```torch.nn.module``` instance.
4
+
`ModelUnderTest` should be a `torch.nn.module` instance.
5
5
6
-
```ModelInputs``` should be a tuple of inputs to the forward function.
6
+
`ModelInputs` should be a tuple of inputs to the forward function.
0 commit comments