Skip to content

GitLab

Commit 41f71f6c authored by Jeremiah Liu's avatar Jeremiah Liu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 366178203
parent fd3e8141
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/nlp/modeling/layers/gaussian_process.py
View file @ 41f71f6c
...@@ -58,7 +58,7 @@ class RandomFeatureGaussianProcess(tf.keras.layers.Layer): ...@@ -58,7 +58,7 @@ class RandomFeatureGaussianProcess(tf.keras.layers.Layer):
gp_kernel_type='gaussian', gp_kernel_type='gaussian',
gp_kernel_scale=1., gp_kernel_scale=1.,
gp_output_bias=0., gp_output_bias=0.,
normalize_input=True, normalize_input=False,
gp_kernel_scale_trainable=False, gp_kernel_scale_trainable=False,
gp_output_bias_trainable=False, gp_output_bias_trainable=False,
gp_cov_momentum=0.999, gp_cov_momentum=0.999,
...@@ -67,7 +67,7 @@ class RandomFeatureGaussianProcess(tf.keras.layers.Layer): ...@@ -67,7 +67,7 @@ class RandomFeatureGaussianProcess(tf.keras.layers.Layer):
use_custom_random_features=True, use_custom_random_features=True,
custom_random_features_initializer=None, custom_random_features_initializer=None,
custom_random_features_activation=None, custom_random_features_activation=None,
l2_regularization=0., l2_regularization=1e-6,
gp_cov_likelihood='gaussian', gp_cov_likelihood='gaussian',
return_gp_cov=True, return_gp_cov=True,
return_random_features=False, return_random_features=False,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment