update model import

pytorch · Jul 6, 2020 · bfbf39f · bfbf39f
1 parent 216784f
commit bfbf39f
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 325 deletions.
diff --git a/examples/pipeline_wavernn/model.py b/examples/pipeline_wavernn/model.py
diff --git a/examples/pipeline_wavernn/wavernn.py b/examples/pipeline_wavernn/wavernn.py
@@ -12,7 +12,7 @@
 from transform import Transform
 from datasets import datasets_ljspeech, collate_factory
 from typing import List
-from model import _WaveRNN
+from torchaudio.models import _WaveRNN
 from torch.utils.data import DataLoader
 from torchaudio.datasets.utils import bg_iterator
 from torch.optim.lr_scheduler import ExponentialLR, ReduceLROnPlateau