Class/Object

org.platanios.tensorflow.api.ops.training.optimizers.schedules

WarmUpExponentialSchedule

Related Docs: object WarmUpExponentialSchedule | package schedules

Permalink

class WarmUpExponentialSchedule extends Schedule

Learning rate schedule that implements a warm-up scheme, similar to the one proposed in [Attention is All You Need (Section 5.3)](https://arxiv.org/pdf/1706.03762.pdf).

For the first warmUpSteps steps the learning rate is multiplied by: exp(log(warmUpFactor) / step) ^ (warmUpSteps - step).

Linear Supertypes
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. WarmUpExponentialSchedule
  2. Schedule
  3. AnyRef
  4. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new WarmUpExponentialSchedule(warmUpSteps: Int, warmUpFactor: Float = 0.01f, name: String = "WarmUpExponentialSchedule")

    Permalink

    warmUpSteps

    Number of warm-up steps.

    warmUpFactor

    Warm-up learning rate scaling factor.

    Attributes
    protected

Value Members

  1. final def !=(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  4. def >>(other: Schedule): ComposedSchedule

    Permalink

    Composes the provided other schedule with this schedule and returns the resulting schedule.

    Composes the provided other schedule with this schedule and returns the resulting schedule.

    Definition Classes
    Schedule
  5. def apply(value: Output, step: Option[variables.Variable]): Output

    Permalink

    Applies the scheduling method to value, the current iteration in the optimization loop is step and returns the result.

    Applies the scheduling method to value, the current iteration in the optimization loop is step and returns the result.

    value

    Value to change based on this schedule.

    step

    Option containing current iteration in the optimization loop, if one has been provided.

    returns

    Potentially modified value.

    Definition Classes
    WarmUpExponentialScheduleSchedule
    Annotations
    @throws( ... )
    Exceptions thrown

    IllegalArgumentException If the scheduling method requires a value for step but the provided option is empty.

  6. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any
  7. def clone(): AnyRef

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  8. def compose(other: Schedule): ComposedSchedule

    Permalink

    Composes this schedule with the provided, other schedule and returns the resulting schedule.

    Composes this schedule with the provided, other schedule and returns the resulting schedule.

    Definition Classes
    Schedule
  9. final def eq(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  10. def equals(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  11. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  12. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any
  13. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  14. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any
  15. val name: String

    Permalink
  16. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  17. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef
  18. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef
  19. def schedule(step: Output, warmUpSteps: Output, warmUpFactor: Output): Output

    Permalink
  20. final def synchronized[T0](arg0: ⇒ T0): T0

    Permalink
    Definition Classes
    AnyRef
  21. def toString(): String

    Permalink
    Definition Classes
    AnyRef → Any
  22. final def wait(): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  23. final def wait(arg0: Long, arg1: Int): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  24. final def wait(arg0: Long): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  25. val warmUpFactor: Float

    Permalink

    Warm-up learning rate scaling factor.

  26. val warmUpSteps: Int

    Permalink

    Number of warm-up steps.

Inherited from Schedule

Inherited from AnyRef

Inherited from Any

Ungrouped