add augment

2026-05-01 11:58:14 +06:00
parent 78519a42b8
commit fad4d68c33
2 changed files with 15 additions and 23 deletions
@@ -213,16 +213,17 @@ def main():

    # ============ 创建数据加载器 ============
    train_loader = create_dataloader(
-        tsv_path=workspace_dir / '.data/train.tsv',
-        audio_dir=workspace_dir / '.data/clips',
+        tsv_path=workspace_dir / '.data/ug/train.tsv',
+        audio_dir=workspace_dir / '.data/ug/clips',
        tokenizer=tokenizer,
        batch_size=CONFIG['batch_size'],
        shuffle=True,
+        augment=True
    )
    
    val_loader = create_dataloader(
-        tsv_path=workspace_dir / '.data/dev.tsv',
-        audio_dir=workspace_dir / '.data/clips',
+        tsv_path=workspace_dir / '.data/ug/dev.tsv',
+        audio_dir=workspace_dir / '.data/ug/clips',
        tokenizer=tokenizer,
        batch_size=CONFIG['batch_size'],
        shuffle=False,