Skip to content

training: expose torch.optim.lr_scheduler.LRScheduler to builder #1027

@trivoldus28

Description

@trivoldus28

Like in

for k in dir(torch.optim):
if k[0].isupper():
builder.register(f"torch.optim.{k}")(getattr(torch.optim, k))

Metadata

Metadata

Assignees

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions