code: def __init__(
      self, checkpoint_wrapper: FairScaleCheckpointWrapper
  ) -> None:
      super().__init__()
      self.ffn = nn.Sequential(
          checkpoint_wrapper.wrap_module(nn.Linear(3, 3)),
          nn.Linear(3, 3),
          checkpoint_wrapper.wrap_module(FeedForwardForTesting()),
          checkpoint_wrapper.wrap_module(nn.Linear(3, 3)),
      )
  
file_path: allennlp/tests/nn/checkpoint/fairscale_checkpoint_wrapper_test.py
imports: import torch
  import torch.nn
  from allennlp.common.params import Params
  from allennlp.common.testing import multi_device
  from allennlp.nn.checkpoint.checkpoint_wrapper import CheckpointWrapper
  from allennlp.nn.checkpoint.fairscale_checkpoint_wrapper import FairScaleCheckpointWrapper
label_desc: Use image node_id 1 to create a new ModuleForTesting object from inherited base classes: nn with example: obj = ModuleForTesting(checkpoint_wrapper)