DiffSpeech / utils /commons /single_thread_env.py
RayeRen's picture
init
d1b91e7
raw
history blame
133 Bytes
import os
os.environ["OMP_NUM_THREADS"] = "1"
os.environ['TF_NUM_INTEROP_THREADS'] = '1'
os.environ['TF_NUM_INTRAOP_THREADS'] = '1'