ksuit.schedules.functional

Functions

linear(step, total_steps)	linearly increasing from [0 to 1]
cosine(step, total_steps)	cosine schedule from [0 to 1]
polynomial(step, total_steps, power)	polynomial schedule from [0 to 1]

Module Contents

ksuit.schedules.functional.linear(step, total_steps)

linearly increasing from [0 to 1]

Parameters:

• step (int)

• total_steps (int)

Return type:

float

ksuit.schedules.functional.cosine(step, total_steps)

cosine schedule from [0 to 1]

Parameters:

• step (int)

• total_steps (int)

Return type:

float

ksuit.schedules.functional.polynomial(step, total_steps, power)

polynomial schedule from [0 to 1] https://pytorch.org/docs/stable/generated/torch.optim.lr_scheduler.PolynomialLR.html

Parameters:

• step (int)

• total_steps (int)

• power (float)

Return type:

float