lr_schedule

Classes

BaseLRSchedule
LinearLRSchedule

Module Contents

class lr_schedule.BaseLRSchedule(cfg)

Parameters:

cfg (dict | float)

classmethod create(cfg)

Parameters:

cfg (dict | float)

Return type:

BaseLRSchedule

func(progress_remaining)

Get the learning rate :param progress_remaining: (float) :return: (float)

Parameters:

progress_remaining (float)

Return type:

float

__call__(progress_remaining)

Parameters:

progress_remaining (float)

Return type:

float

class lr_schedule.LinearLRSchedule(cfg)

Bases: BaseLRSchedule

Parameters:

cfg (dict | float)

func(progress_remaining)

Get the current learning rate depending on remaining progress. :param progress_remaining: (float) :return: (float)

Parameters:

progress_remaining (float)

Return type:

float