diff options
Diffstat (limited to 'tensorflow')
-rw-r--r-- | tensorflow/python/keras/_impl/keras/layers/core.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/keras/_impl/keras/layers/core.py b/tensorflow/python/keras/_impl/keras/layers/core.py index 87b997232e..f64174a23f 100644 --- a/tensorflow/python/keras/_impl/keras/layers/core.py +++ b/tensorflow/python/keras/_impl/keras/layers/core.py @@ -836,7 +836,7 @@ class Dense(Layer): super(Dense, self).__init__( activity_regularizer=regularizers.get(activity_regularizer), **kwargs) - self.units = units + self.units = int(units) self.activation = activations.get(activation) self.use_bias = use_bias self.kernel_initializer = initializers.get(kernel_initializer) |