Fix `too many open files`

This commit is contained in:
Eren Gölge 2021-12-08 15:16:16 +00:00
parent df0d58bf09
commit c0746f23df
1 changed files with 4 additions and 0 deletions

View File

@ -11,6 +11,10 @@ from torch.utils.data import Dataset
from TTS.tts.utils.data import prepare_data, prepare_stop_target, prepare_tensor
from TTS.utils.audio import AudioProcessor
# to prevent too many open files error as suggested here
# https://github.com/pytorch/pytorch/issues/11201#issuecomment-421146936
torch.multiprocessing.set_sharing_strategy('file_system')
def _parse_sample(item):
language_name = None