code: def gen_extra_context(self, model):
      super().gen_extra_context()
      assert (
          model is not None
      ), "Must include a model for total number of parameter calculation"
      self.total_param_count = sum(
          p.numel() for p in model.parameters() if p.requires_grad
      )
  
imports: from collections import defaultdict
  from typing import Dict, List
  import numpy
  from pytext.common.constants import Stage
  from pytext.data.tensorizers import Tensorizer
  from pytext.metric_reporters.channel import ConsoleChannel
  from pytext.metric_reporters.compositional_metric_reporter import CompositionalMetricReporter
  from pytext.metric_reporters.seq2seq_compositional import CompositionalSeq2SeqFileChannel, Seq2SeqCompositionalMetricReporter
  from pytext.metrics.intent_slot_metrics import FramePredictionPair
  from pytext.metrics.mask_metrics import compute_masked_metrics, compute_nas_masked_metrics
  from .compositional_utils import extract_beam_subtrees, filter_invalid_beams
label_desc: Use image node_id 3 for calling the NASMaskedSeq2SeqCompositionalMetricReporter obj's underlying member method code with example usage: obj.gen_extra_context(model) without return types