Registry / ai-ml / prodigy-plus-schedule-free

prodigy-plus-schedule-free

JSON →
library2.0.1pypypiunverified

Automatic learning rate optimizer combining Prodigy's adaptive LR with Schedule-Free's constant-parameter interpolation. Version 2.0.1 improved weight decay handling. Active development.

pip install prodigy-plus-schedule-free
INSTALL
IMPORT
SIG · PRODIGY-PLUS-SCHED
P
prodigy-plus-schedule-free
ai-mlpythonv2.0.1
Install
76.2s avg
Import
—
Disk
5140MB
Pass rate
5/ 10
Env Coverage5 / 10
glibc
3.9–3.13
musl
3.9–3.13
Install & Compatibility
Where this runs
tested against v2.0.1 · pip install
no network on importno background threads
Install × environment matrix
Each cell = how many times install + import succeeded across repeated harness runs. Partial = flaky.
glibc = Debian/Ubuntu slim · musl = Alpine Linux
musl
py 3.10–3.95 runs
build_error
glibc
py 3.10–3.95 runs
installs and imports cleanly · install 76.2s · import 0.000s · 4710.4MB
5140MB installed
● package 5140MB
Code
Verified usage

Verified import paths — ran on the pinned version, not inferred.

ProdigyPlusScheduleFree
✓ from prodigy_plus_schedule_free import ProdigyPlusScheduleFree
✗ from prodigy_plus_schedule_free import ProdigyPlusScheduleFree
Common mistake: using 'prodigyplus' instead of 'prodigy_plus_schedule_free'.

Basic usage: instantiate optimizer, call .train() before training loop, step normally.

import torch from prodigy_plus_schedule_free import ProdigyPlusScheduleFree model = torch.nn.Linear(10, 2) optimizer = ProdigyPlusScheduleFree(model.parameters(), lr=1.0) optimizer.train() for data, target in [(torch.randn(10), torch.tensor(1))]: optimizer.zero_grad() loss = torch.nn.functional.cross_entropy(model(data), target.unsqueeze(0)) loss.backward() optimizer.step()
Debug
Known issues
breakingIn v2.0.0, the import path changed from 'prodigyplus_schedulefree' to 'prodigy_plus_schedule_free'. Old imports will break.
fix
Update import: 'from prodigy_plus_schedule_free import ProdigyPlusScheduleFree'.
affects: <2.0.0
gotchaYou must call .train() at the start of each training loop and .eval() for evaluation to ensure correct parameter interpolation.
fix
Always switch modes: optimizer.train() before training, optimizer.eval() before inference.
affects: all
deprecatedParameter 'weight_decay' had a bug in v1.x where it was applied incorrectly. Use v2.0.1+ for correct weight decay.
fix
Upgrade to >=2.0.1. If you cannot upgrade, avoid using weight_decay or implement manually.
affects: <2.0.1
Upgrade
Version history
2.0.1latest on PyPI · released Sep 27, 2025
Audit
Dependencies
torchrequiredPyTorch is required for optimizers and tensors.
Agent activity
5 hits · last 30 days
node
4
Bingbot
1
Resources
prodigy-plus-schedule-free — pip install prodigy-plus-schedule-free · libregistry