code: def tensor_sizes(state):
      size_count = collections.defaultdict(int)
      for key, value in state.items():
          if key.startswith("fc."):
              continue  # "head" for torchvision
          size_count[value.size()] += 1
      return dict(size_count)
  
imports: import collections
  import unittest
  import torch
  import torchvision.models
  from classy_vision.generic.util import get_torch_version
  from classy_vision.models import build_model, ResNeXt
  from test.generic.utils import compare_model_state
label_desc: Use image node_id 1 for calling the TestTorchvisionEquivalence obj's underlying member method code with example usage: obj.tensor_sizes(state) and returns: dict