Spaces:
				
			
			
	
			
			
		Sleeping
		
	
	
	
			
			
	
	
	
	
		
		
		Sleeping
		
	Update train.py
Browse files
    	
        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 = {
         | 
