org.nd4j.linalg.schedule

Class RampSchedule

java.lang.Object

org.nd4j.linalg.schedule.RampSchedule

All Implemented Interfaces:

Serializable, Cloneable, ISchedule

public class RampSchedule
extends Object
implements ISchedule

A "Wrapper" schedule that ramps up from 1/numIter * baseLR to baseLR over numIter iterations. The base learning rate is determined by the underlying ISchedule, as a function of time. This can be used to provide a slow start, for use cases such as transfer learning.

Author:

Alex Black

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected ISchedule	baseSchedule
protected int	numIter

Constructor Summary

Constructors

Constructor and Description
RampSchedule(ISchedule baseSchedule, int numIter)

Method Summary

Modifier and Type	Method and Description
ISchedule	clone()
double	valueAt(int iteration, int epoch)

Methods inherited from class java.lang.Object

equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

baseSchedule

protected final ISchedule baseSchedule

numIter

protected final int numIter

Constructor Detail

RampSchedule

public RampSchedule(ISchedule baseSchedule,
                    int numIter)

Method Detail

valueAt

public double valueAt(int iteration,
                      int epoch)

Specified by:

valueAt in interface ISchedule

Parameters:

iteration - Current iteration number. Starts at 0

epoch - Current epoch number. Starts at 0

Returns:

Value at the current iteration/epoch for this schedule

clone

public ISchedule clone()

Specified by:

clone in interface ISchedule

Overrides:

clone in class Object