Skip to content

[FR] rename LR scheduler base class _LRScheduler to be non-private #61232

@ssnl

Description

@ssnl

🚀 Feature

Rename _LRScheduler to something like LRScheduler without underscore.

Motivation

Allows for writing typed code much better. We have torch.optim.Optimizer but no public base class for learning rate schedulers.

cc @ezyang @gchanan @zou3519 @vincentqb @jbschlosser @albanD @iramazanli

Metadata

Metadata

Assignees

Labels

actionableenhancementNot as big of a feature, but technically not a bug. Should be easy to fixhigh prioritymodule: optimizerRelated to torch.optimtriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

Type

No type

Projects

Status

Done

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions