code: def __init__(
      self,
      model_size: int,
      learning_rate: float,
      warmup: int,
      step_factor: int,
      optimizer,
  ):
      self.step_factor = step_factor
      self.optimizer = optimizer
      self.warmup = warmup
      self.learning_rate = learning_rate
      self.model_size = model_size
      self._rate = 0
  
file_path: labml/helpers/labml_helpers/optimizer.py
imports: from typing import Tuple
  import torch
  from labml import tracker
  from labml.configs import BaseConfigs, option, meta_config
label_desc: Use image node_id 1 to create a new NoamOpt object  with example: obj = NoamOpt(model_size, learning_rate, warmup, step_factor, optimizer)