class_docstr: Test cases for ResNet v1.5 model definition.
docstr: Tests ResNet V1.5 model definition and output (variables).
code: def test_resnet_model(self):
      """Tests ResNet V1.5 model definition and output (variables)."""
      rng = jax.random.key(0)
      model_def = models.ResNet50(num_classes=10, dtype=jnp.float32)
      variables = model_def.init(
          rng, jnp.ones((8, 224, 224, 3), jnp.float32)
      )
  
      self.assertLen(variables, 2)
      # Resnet50 model will create parameters for the following layers:
      #   conv + batch_norm = 2
      #   BottleneckResNetBlock in stages: [3, 4, 6, 3] = 16
      #   Followed by a Dense layer = 1
      self.assertLen(variables["params"], 19)
  
file_path: flax/examples/imagenet/models_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  from jax import numpy
  import models
label_desc: Use image node_id 1 for calling the ResNetTest obj's underlying member method code with example usage: obj.test_resnet_model() without return types