Implements cosine decay with linear warmup #15
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This pull request introduces a new cosine decay schedule with linear warmup to the
Polaris.Schedules
module and adds corresponding tests to ensure its functionality.New Feature: Cosine Decay Schedule with Linear Warmup
lib/polaris/schedules.ex
: Added thewarmup_cosine_decay
function, which implements a learning rate schedule that first increases linearly during a warmup period and then follows a cosine decay pattern. This function includes detailed documentation and options for customization.Tests for New Feature
test/polaris/schedules_test.exs
: Added a series of tests for thewarmup_cosine_decay
function to verify its behavior, including tests for correct function arity, linear warmup phase, cosine decay phase, and respect for thealpha
parameter.