code: def test_nested_wrapped_weight_init(self, config):
      model_init_fn = functools.partial(
          model_init_and_apply_custom_weight_init, NestedWrappedModule
      )
      test_fn = functools.partial(
          self._test_identical_outputs, model_init_fn, config, lr=0.01
      )
      spawn_and_init(test_fn)
  
file_path: fairscale/tests/nn/data_parallel/test_fsdp_apply.py
imports: import functools
  import unittest
  from parameterized import parameterized
  import pytest
  import torch.nn
  from fairscale.internal import torch_version
  from .test_fsdp import CONFIG_OPTIONS, DistributedTest, NestedWrappedModule, TransformerWithSharedParams, rename_test, spawn_and_init
label_desc: Use image node_id 2 for calling the TestApply obj's underlying member method code with example usage: obj.test_nested_wrapped_weight_init(config) without return types