Class FixedPerVarTracker.Builder

    • Method Detail

      • setDefaultValue

        public FixedPerVarTracker.Builder setDefaultValue​(float value)
        Set the default learning rate.
        Parameters:
        value - the default learning rate
        Returns:
        builder
      • put

        public FixedPerVarTracker.Builder put​(java.lang.String parameterId,
                                              float value)
        Add a kv pair of parameter and its learning rate.
        Parameters:
        parameterId - the parameter id
        value - the default learning rate
        Returns:
        builder
      • putAll

        public FixedPerVarTracker.Builder putAll​(java.util.Map<java.lang.String,​java.lang.Float> valueMap)
        Add kv pairs of parameter and its learning rate.
        Parameters:
        valueMap - stores parameterId and learning rate
        Returns:
        builder