Add Inception V4 model to examples (#86)

Summary:
Pull Request resolved: #86

The model is from pypi/timm.

Reviewed By: guangy10

Differential Revision: D48524577

fbshipit-source-id: 8d8eb302a9ca5cc66ba8056391a1826cbb2b7a61

Author: kirklandsign

examples/models/TARGETS

@@ -9,6 +9,7 @@ python_library(
     deps = [
         "//caffe2:torch",
         "//executorch/examples/models/inception_v3:ic3_export",
+        "//executorch/examples/models/inception_v4:ic4_export",
         "//executorch/examples/models/mobilenet_v2:mv2_export",
         "//executorch/examples/models/mobilenet_v3:mv3_export",
         "//executorch/examples/models/torchvision_vit:vit_export",

examples/models/inception_v4/TARGETS

@@ -0,0 +1,14 @@
+load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
+
+python_library(
+    name = "ic4_export",
+    srcs = [
+        "__init__.py",
+        "export.py",
+    ],
+    base_module = "executorch.examples.models.inception_v4",
+    deps = [
+        "fbsource//third-party/pypi/timm:timm",
+        "//caffe2:torch",
+    ],
+)

examples/models/inception_v4/__init__.py

@@ -0,0 +1,11 @@
+# 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.
+
+from .export import InceptionV4Model
+
+__all__ = [
+    InceptionV4Model,
+]

examples/models/inception_v4/export.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.
+
+import logging
+
+import torch
+from timm.models import inception_v4
+
+FORMAT = "[%(filename)s:%(lineno)s] %(message)s"
+logging.basicConfig(format=FORMAT)
+
+
+# will refactor this in a separate file.
+class InceptionV4Model:
+    def __init__(self):
+        pass
+
+    @staticmethod
+    def get_model():
+        logging.info("loading inception_v4 model")
+        m = inception_v4(pretrained=True)
+        logging.info("loaded inception_v4 model")
+        return m
+
+    @staticmethod
+    def get_example_inputs():
+        return (torch.randn(3, 299, 299).unsqueeze(0),)

examples/models/models.py

@@ -108,6 +108,12 @@ def gen_inception_v3_model_and_inputs() -> Tuple[torch.nn.Module, Any]:
     return InceptionV3Model.get_model(), InceptionV3Model.get_example_inputs()
 
 
+def gen_inception_v4_model_and_inputs() -> Tuple[torch.nn.Module, Any]:
+    from ..models.inception_v4 import InceptionV4Model
+
+    return InceptionV4Model.get_model(), InceptionV4Model.get_example_inputs()
+
+
 MODEL_NAME_TO_MODEL = {
     "mul": lambda: (MulModule(), MulModule.get_example_inputs()),
     "linear": lambda: (LinearModule(), LinearModule.get_example_inputs()),
@@ -118,4 +124,5 @@ def gen_inception_v3_model_and_inputs() -> Tuple[torch.nn.Module, Any]:
     "vit": gen_torchvision_vit_model_and_inputs,
     "w2l": gen_wav2letter_model_and_inputs,
     "ic3": gen_inception_v3_model_and_inputs,
+    "ic4": gen_inception_v4_model_and_inputs,
 }