def efficientnet_tpu_params(model_name):
  """Get efficientnet params based on model name."""
  params_dict = {
      # (width_coefficient, depth_coefficient, resolution, dropout_rate,
      #  se_coefficient)
      'efficientnet-tpu-b0': (1.0, 1.0, 224, 0.2, 4),
      'efficientnet-tpu-b1': (1.0, 1.1, 240, 0.2, 2),
      'efficientnet-tpu-b2': (1.1, 1.2, 260, 0.3, 1),
      'efficientnet-tpu-b3': (1.2, 1.4, 300, 0.3, 1),
      'efficientnet-tpu-b4': (1.4, 1.8, 380, 0.4, 1),
      'efficientnet-tpu-b5': (1.6, 2.2, 456, 0.4, 1),
      'efficientnet-tpu-b6': (1.8, 2.6, 528, 0.5, 1),
      'efficientnet-tpu-b7': (2.0, 3.1, 600, 0.5, 1),
  }
  return params_dict[model_name]