update imports for util refactoring

pull/10/head
erogol 2020-05-12 13:46:32 +02:00
parent 67420eeb86
commit 2d9dcd60ba
1 changed files with 2 additions and 1 deletions

View File

@ -9,7 +9,8 @@ import torch.distributed as dist
from torch.utils.data.sampler import Sampler
from torch.autograd import Variable
from torch._utils import _flatten_dense_tensors, _unflatten_dense_tensors
from TTS.utils.generic_utils import load_config, create_experiment_folder
from TTS.utils.io import load_config
from TTS.utils.generic_utils import create_experiment_folder
class DistributedSampler(Sampler):