I’m doing something like this
Move to memory
common_voice_train = common_voice_train.map(lambda x:x,keep_in_memory=True)
Move to disk
common_voice_train = common_voice_train.map(lambda x:x,keep_in_memory=False)