code: def __init__(self):
      super().__init__()
      self.mod = TestModuleWithoutFlops()
      # add a conv module; this shouldn't impact the FLOPs since we define
      # self.flops()
      self.conv = nn.Conv2d(3, 3, (2, 2))
  
imports: import unittest
  import torch
  import torch.nn
  from classy_vision.generic.profiler import compute_activations, compute_flops, count_params, get_shape
  from classy_vision.models import build_model
  from test.generic.config_utils import get_test_model_configs
label_desc: Use image node_id 1 to create a new TestModuleWithFlops object from inherited base classes: nn with example: obj = TestModuleWithFlops()