Add resnet18 test case to OSS

Differential Revision: D74206440

Pull Request resolved: #10705

Author: mcremon-meta

examples/cadence/models/resnet18.py

@@ -0,0 +1,30 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+# Example script for exporting simple models to flatbuffer
+
+import logging
+
+import torch
+
+from executorch.backends.cadence.aot.ops_registrations import *  # noqa
+
+
+from executorch.backends.cadence.aot.export_example import export_model
+from torchvision.models import resnet18, ResNet18_Weights
+
+
+FORMAT = "[%(levelname)s %(asctime)s %(filename)s:%(lineno)s] %(message)s"
+logging.basicConfig(level=logging.INFO, format=FORMAT)
+
+
+if __name__ == "__main__":
+
+    model = resnet18(weights=ResNet18_Weights.DEFAULT)
+    model.eval()
+    example_inputs = (torch.randn(1, 3, 64, 64),)
+
+    export_model(model, example_inputs)