class_docstr: Used for SGD Lars
code: def __init__(self, optimizer, max_iter, last_step=-1):
      super(SquaredLR, self).__init__(
          optimizer, lambda s: (1 - s / (max_iter + 1)) ** 2, last_step
      )
  
imports: import sys
  from torch.optim import lr_scheduler
  from omegaconf import OmegaConf
  from omegaconf.dictconfig import DictConfig
  import logging
  from torch.optim.lr_scheduler import LambdaLR
  from torch_points3d.utils.enums import SchedulerUpdateOn
label_desc: Use image node_id 1 to create a new SquaredLR object from inherited base classes: LambdaStepLR with example: obj = SquaredLR(optimizer, max_iter, last_step)