gpflux.layers.trackable_layer
#
Utility layer that tracks variables in tf.Module
.
Module Contents#
- class TrackableLayer(trainable=True, name=None, dtype=None, dynamic=False, **kwargs)[source]#
Bases:
tf.keras.layers.Layer
With the release of TensorFlow 2.5, our TrackableLayer workaround is no longer needed. See Prowler-io/gpflux#189. Will be removed in GPflux version 1.0.0