def get_long_data_subset(dataset, size=500):
    """Get a subset of desired size by sorting based on src_lengths descending"""
    return sort_dataset(dataset, reverse=True)[:size]
Get a subset of desired size by sorting based on src_lengths descending
examples/speech_to_speech/benchmarking/data_utils.py