class_docstr: Base test class and used for CPU case.
docstr: Make sure numel of params are the same after flatten.
docstr_tok: ["Make","sure","numel","of","params","are","the","same","after","flatten","."]
code: def _test_num_params(self, module):
      """Make sure numel of params are the same after flatten."""
      ref_num_params = sum(p.numel() for p in module.parameters())
  
      flat_module = FlattenParamsWrapper(module)
      flat_num_params = sum(p.numel() for p in flat_module.parameters())
  
      assert ref_num_params == flat_num_params
      assert flat_num_params == flat_module.flat_param.numel()
  
imports: from collections import OrderedDict
  import unittest
  import torch
  from fairscale.fair_dev.testing.testing import objects_are_equal
  from fairscale.nn import FlattenParamsWrapper
label_desc: Use image node_id 10 for calling the TestFlattenParams obj's underlying member method code with example usage: obj._test_num_params(module) without return types