code: def __init__(self, lr_scheduler, rescale_grad):
      self.lr_scheduler = lr_scheduler
      self.rescale_grad = rescale_grad
      self.momentum = 0.9
      self.wd = 5e-4
  
file_path: insightface/recognition/partial_fc/mxnet/optimizer.py
imports: import horovod.mxnet
  import mxnet
  from mxnet import nd
label_desc: Use image node_id 1 to create a new MemoryBankSGDOptimizer object from inherited base classes: object with example: obj = MemoryBankSGDOptimizer(lr_scheduler, rescale_grad)