Internal change

PiperOrigin-RevId: 389675343
SemaSandbox · Aug 9, 2021 · 66ddb0c · 66ddb0c
1 parent 468f201
commit 66ddb0c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/official/vision/beta/modeling/backbones/resnet.py b/official/vision/beta/modeling/backbones/resnet.py
@@ -32,6 +32,12 @@
 # Each element in the block configuration is in the following format:
 # (block_fn, num_filters, block_repeats)
 RESNET_SPECS = {
+    10: [
+        ('residual', 64, 1),
+        ('residual', 128, 1),
+        ('residual', 256, 1),
+        ('residual', 512, 1),
+    ],
     18: [
         ('residual', 64, 2),
         ('residual', 128, 2),

diff --git a/official/vision/beta/modeling/backbones/resnet_test.py b/official/vision/beta/modeling/backbones/resnet_test.py
@@ -28,6 +28,7 @@
 class ResNetTest(parameterized.TestCase, tf.test.TestCase):
 
   @parameterized.parameters(
+      (128, 10, 1),
       (128, 18, 1),
       (128, 34, 1),
       (128, 50, 4),
@@ -38,6 +39,7 @@ def test_network_creation(self, input_size, model_id,
                             endpoint_filter_scale):
     """Test creation of ResNet family models."""
     resnet_params = {
+        10: 4915904,
         18: 11190464,
         34: 21306048,
         50: 23561152,