Class BasePretrainNetwork

    • Method Detail

      • isPretrainParam

        public boolean isPretrainParam​(String paramName)
        Description copied from class: Layer
        Is the specified parameter a layerwise pretraining only parameter?
        For example, visible bias params in an autoencoder (or, decoder params in a variational autoencoder) aren't used during supervised backprop.
        Layers (like DenseLayer, etc) with no pretrainable parameters will return false for all (valid) inputs.
        Specified by:
        isPretrainParam in interface TrainingConfig
        Overrides:
        isPretrainParam in class FeedForwardLayer
        Parameters:
        paramName - Parameter name/key
        Returns:
        True if the parameter is for layerwise pretraining only, false otherwise