def unet_params(model_name):
  """Get U-Net params based on model name."""
  params_dict = {
      # (width_coefficient, depth_coefficient, resolution, dropout_rate)
    #   'unet-b0': (1.0, 1.0, 224, 0.2),
    #   'unet-b1': (1.0, 1.1, 240, 0.2),
    #   'unet-b2': (1.1, 1.2, 260, 0.3),
    #   'unet-b3': (1.2, 1.4, 300, 0.3),
    #   'unet-b4': (1.4, 1.8, 380, 0.4),
    #   'unet-b5': (1.6, 2.2, 456, 0.4)
        'unet-w0': (1.0, 1.0, 300, 0.2),
        'unet-w1': (1.3, 1.0, 300, 0.2),
        'unet-w2': (1.6, 1.0, 300, 0.2),
        'unet-w3': (1.9, 1.0, 300, 0.2),
        'unet-w4': (2.1, 1.0, 300, 0.2),
        'unet-r0': (1.0, 1.0, 300, 0.2),
        'unet-r1': (1.0, 1.0, 320, 0.2),
        'unet-r2': (1.0, 1.0, 360, 0.2),
        'unet-r3': (1.0, 1.0, 420, 0.2),
        'unet-r4': (1.0, 1.0, 500, 0.2),
        'unet-v1': (1.3, 1.0, 320, 0.2),
  }
  return params_dict[model_name]