    def get_gather_args(self):
        """obtain the args for all_gather
        Returns: dict.

        """
        return EasyDict({"world_size": get_world_size(), "rank": get_rank()})
obtain the args for all_gather
Returns: dict.
videomamba/video_mm/models/criterions.py