keras-one-cycle icon indicating copy to clipboard operation
keras-one-cycle copied to clipboard

Documentation has different initialization

Open rmcconke opened this issue 4 years ago • 0 comments

The initialization in clr.py is: def __init__(self, num_samples, batch_size, max_lr, end_percentage=0.1, scale_percentage=None, maximum_momentum=0.95, minimum_momentum=0.85, verbose=True)

The initialization in the documentation is: lr_manager = OneCycleLR(num_samples, num_epoch, batch_size, max_lr end_percentage=0.1, scale_percentage=None, maximum_momentum=0.95, minimum_momentum=0.85)

This causes unexpected behaviour in the function compute_lr.

rmcconke avatar Feb 14 '21 23:02 rmcconke