def get_policy(self):
        """
        Return a policy for wrapping different submodules of a model with FSDP wrapper.
        """

        Return a policy for wrapping different submodules of a model with FSDP wrapper.
        
coremltools/optimize/torch/_utils/fsdp_utils.py