training: expose `torch.optim.lr_scheduler.LRScheduler` to builder

[trivoldus28]

Like in

zetta_utils/zetta_utils/convnet/architecture/primitives.py

Lines 25 to 27 in bc2edab

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

[trivoldus28]

I found this to work

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