Class FakeDDP

tests/test_model/test_base_module.py:225–229

        self.assertTrue((ori_layer_bias != model.linear.linear.bias).any())

        class FakeDDP(nn.Module):

            def __init__(self, module) -> None:
                super().__init__()
                self.module = module

        # Test initialization of nested modules in DDPModule which define
        # `init_weights`.