def get_features(self, batch):
    (metas, [video_features, audio_features]) = self._get_features(batch)
    if self.distributed:
        i = du.get_rank()
        total = du.get_world_size()
        metas = metas[i::total]
        video_features = video_features[i::total]
        audio_features = audio_features[i::total]
    return (metas, [video_features, audio_features])
def get_features(self, batch):
    (metas, [video_features, audio_features]) = self._get_features(batch)
    if self.distributed:
        i = du.get_rank()
        total = du.get_world_size()
        metas = metas[i::total]
        video_features = video_features[i::total]
        audio_features = audio_features[i::total]
    return (metas, [video_features, audio_features])