code: def compute_amax(model, **kwargs):
      for name, module in model.named_modules():
          if isinstance(module, quant_nn.TensorQuantizer):
              if module._calibrator is not None:
                  if isinstance(
                      module._calibrator, calib.MaxCalibrator
                  ):
                      module.load_calib_amax()
                  else:
                      module.load_calib_amax(**kwargs)
              print(f"{name:40}: {module}")
      if can_gpu:
          model.cuda()
  
imports: from argparse import ArgumentParser
  import torch
  from omegaconf import open_dict
  from nemo.collections.asr.models import EncDecCTCModel
  from nemo.utils import logging
label_desc: Use image node_id 2 for calling a global function with example usage: compute_amax(model) without return types