Function set_trainable

gpflow/utilities/misc.py:57–66

Set trainable flag for all :class:`tf.Variable`\ s and :class:`gpflow.Parameter`\ s in a :class:`tf.Module` or collection of :class:`tf.Module`\ s.

(model: Union[tf.Module, Iterable[tf.Module]], flag: bool)

def set_trainable(model: Union[tf.Module, Iterable[tf.Module]], flag: bool) -> None:
    """
    Set trainable flag for all :class:`tf.Variable`\ s and :class:`gpflow.Parameter`\ s in a
    :class:`tf.Module` or collection of :class:`tf.Module`\ s.
    """
    modules = [model] if isinstance(model, tf.Module) else model

    for mod in modules:
        for variable in mod.variables:
            variable._trainable = flag


def is_variable(t: TensorData) -> bool: