mllam/neural-lam

Learning rate scheduler

Open

#97 opened on Dec 10, 2024

View on GitHub
 (8 comments) (0 reactions) (1 assignee)Python (276 forks)auto 404
enhancementgood first issue

Repository metrics

Stars
 (282 stars)
PR merge metrics
 (PR metrics pending)

Description

Learning rate scheduling is something useful that we should have as an option. As setting up a way to configure arbitrary pytorch schedulers becomes over-complicated, we should start by just adding in options for cosine schedule. @ErikLarssonDev has a prototype for this.

Contributor guide