Skip to content
This repository was archived by the owner on May 21, 2025. It is now read-only.
This repository was archived by the owner on May 21, 2025. It is now read-only.

Small bug in create tower #361

@shanuv

Description

@shanuv

In lines 70 and 76 of tensorflow_ranking/python/keras/layers.py, the code snippet should have **kwargs as input to tf.keras.layers.Dense

model = tf.keras.Sequential(name=name)
  # Input batch normalization.
  if input_batch_norm:
    model.add(tf.keras.layers.BatchNormalization(momentum=batch_norm_moment))
  for layer_width in hidden_layer_dims:
    model.add(tf.keras.layers.Dense(units=layer_width), **kwargs)
    if use_batch_norm:
      model.add(tf.keras.layers.BatchNormalization(momentum=batch_norm_moment))
    model.add(tf.keras.layers.Activation(activation=activation))
    if dropout:
      model.add(tf.keras.layers.Dropout(rate=dropout))
  model.add(tf.keras.layers.Dense(units=output_units), **kwargs)
  return model

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions