[proto-runner] Add resnext to the list of supported models.

zakk0610 · Aug 10, 2018 · 09b9117 · 09b9117
1 parent 913a7c2
commit 09b9117
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/utils/caffe2_pb_runner.py b/utils/caffe2_pb_runner.py
@@ -98,6 +98,12 @@ def mode_128to127 (x): return x-128
         28,
         1,
     ),
+    resnext=Model(
+        "data",
+        mode_0to1,
+        224,
+        3,
+    ),
 )
 
 MODEL = args.model_name