Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix passing of keyword args to Dense layers in create_tower #339

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Commits on Dec 8, 2022

  1. Fix passing of keyword args to Dense layers in create_tower

    Current behavior: kwargs are passed to tf.keras.Sequential.add, so they
    are not passed on to tf.keras.layers.Dense as intended. For example,
    when passing `use_bias=False` to create_tower with the kwarg name
    `kernel_regularizer`, it throws an exception:
    
    Traceback (most recent call last):
      File "/Users/brussell/development/ranking/tensorflow_ranking/python/keras/layers_test.py", line 33, in test_create_tower_with_kwargs
        tower = layers.create_tower([3, 2, 1], 1, activation='relu', use_bias=False)
      File "/Users/brussell/development/ranking/tensorflow_ranking/python/keras/layers.py", line 70, in create_tower
        model.add(tf.keras.layers.Dense(units=layer_width), **kwargs)
      File "/usr/local/anaconda3/lib/python3.9/site-packages/tensorflow/python/trackable/base.py", line 205, in _method_wrapper
        result = method(self, *args, **kwargs)
      File "/usr/local/anaconda3/lib/python3.9/site-packages/keras/utils/traceback_utils.py", line 61, in error_handler
        return fn(*args, **kwargs)
    TypeError: add() got an unexpected keyword argument 'use_bias'
    test_create_tower_with_kwargs
    
    Fix: This PR fixes the behavior by shifting the closing paren of
    tf.keras.layers.Dense to the correct location.
    b4russell committed Dec 8, 2022
    Configuration menu
    Copy the full SHA
    1400f70 View commit details
    Browse the repository at this point in the history