Update __init__.py

Chris-hughes10 · web-flow · commit 211d942c59d4 · 2025-02-18T15:05:20.000Z
diff --git a/pytorch_accelerated/schedulers/__init__.py b/pytorch_accelerated/schedulers/__init__.py
@@ -1,4 +1,5 @@
 from pytorch_accelerated.schedulers.cosine_scheduler import CosineLrScheduler
+from pytorch_accelerated.schedulers.wsd_scheduler import WSDLrScheduler
 from pytorch_accelerated.schedulers.scheduler_base import (
     StatefulSchedulerBase,
     SchedulerBase,

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`from pytorch_accelerated.schedulers.cosine_scheduler import CosineLrScheduler`
	`2`	`+from pytorch_accelerated.schedulers.wsd_scheduler import WSDLrScheduler`
`2`	`3`	`from pytorch_accelerated.schedulers.scheduler_base import (`
`3`	`4`	`StatefulSchedulerBase,`
`4`	`5`	`SchedulerBase,`