make a change on args

Author: yistLin

README.md

@@ -14,7 +14,7 @@ python preprocess.py VCTK-Corpus LibriTTS/train-clean-100 preprocessed
 ### Train from scratch
 
 ```bash
-python train.py preprocessed preprocessed/metadata.json
+python train.py preprocessed
 ```
 
 ### Generate waveforms

train.py

@@ -23,7 +23,6 @@ def parse_args():
     """Parse command-line arguments."""
     parser = ArgumentParser()
     parser.add_argument("data_dir", type=str)
-    parser.add_argument("metadata_path", type=str)
     parser.add_argument("--n_workers", type=int, default=8)
     parser.add_argument("--save_dir", type=str, default=".")
     parser.add_argument("--comment", type=str)
@@ -50,7 +49,6 @@ def parse_args():
 
 def main(
     data_dir,
-    metadata_path,
     n_workers,
     save_dir,
     comment,
@@ -75,7 +73,11 @@ def main(
 
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     dataset = VocoderDataset(
-        data_dir, metadata_path, frames_per_sample, frames_per_slice, bits
+        data_dir,
+        Path(data_dir) / "metadata.json",
+        frames_per_sample,
+        frames_per_slice,
+        bits,
     )
     lengths = [
         trainlen := int((1 - valid_ratio) * len(dataset)),