Update vocoder_trainer.py

open-mmlab · Jul 2, 2024 · 3ba040f · 3ba040f
1 parent 3ad778e
commit 3ba040f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models/vocoders/vocoder_trainer.py b/models/vocoders/vocoder_trainer.py
@@ -16,8 +16,8 @@
 from torch.utils.data import DataLoader
 from tqdm import tqdm
 
-from models.codec.codec_dataset import VocoderConcatDataset
-from models.codec.codec_sampler import build_samplers
+from models.vocoders.vocoder_dataset import VocoderConcatDataset
+from models.vocoders.vocoder_sampler import build_samplers
 
 
 class VocoderTrainer: