Ravi-9 commited on
Commit
06aa4a6
·
verified ·
1 Parent(s): f488928

Update train.py

Browse files
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -1,7 +1,7 @@
1
  import torch
2
  from torch.utils.data import DataLoader
3
  from transformers import AdamW, get_linear_schedule_with_warmup
4
- from utils.dataset import BDTtsDataset
5
  from inference import tts # reuse your model
6
 
7
  training_config = {
 
1
  import torch
2
  from torch.utils.data import DataLoader
3
  from transformers import AdamW, get_linear_schedule_with_warmup
4
+ # from utils.dataset import BDTtsDataset
5
  from inference import tts # reuse your model
6
 
7
  training_config = {