From 603c8bb1a9b89006e7e0c1704d8cf518bcd4758d Mon Sep 17 00:00:00 2001 From: Matthias Cremon Date: Mon, 5 May 2025 18:43:30 -0700 Subject: [PATCH] Add resnet18 test case to OSS Summary: As titled. Differential Revision: D74206440 --- examples/cadence/models/resnet18.py | 30 +++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 examples/cadence/models/resnet18.py diff --git a/examples/cadence/models/resnet18.py b/examples/cadence/models/resnet18.py new file mode 100644 index 00000000000..ebf0533bd84 --- /dev/null +++ b/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)