Browse Source

更新 'train_cload.py'

master
lwj 1 month ago
parent
commit
6cd04adae2
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      train_cload.py

+ 4
- 4
train_cload.py View File

@@ -61,11 +61,11 @@ parser.add_argument('--run_distribute', type=bool, default=True,
help='Sample rate of audio file')
parser.add_argument('--data_batch_size', type=int, default=3,
help='Sample rate of audio file')
parser.add_argument('--train', type=str, default='/home/work/user-job-dir/inputs/data_json/tr',
parser.add_argument('--train', type=str, default='/home/work/user-job-dir/inputs/data_json/tt',
help='Sample rate of audio file')
parser.add_argument('--valid', type=str, default="/home/work/user-job-dir/inputs/data_json/tr",
parser.add_argument('--valid', type=str, default="/home/work/user-job-dir/inputs/data_json/tt",
help='Sample rate of audio file')
parser.add_argument('--test', type=str, default="/home/work/user-job-dir/inputs/data_json/tr",
parser.add_argument('--test', type=str, default="/home/work/user-job-dir/inputs/data_json/tt",
help='Sample rate of audio file')
parser.add_argument('--lr', type=float, default=5e-4,
help='Sample rate of audio file')
@@ -169,7 +169,7 @@ def preprocess_one_dir(in_dir, out_dir, out_filename, sample_rate=8000):

def preprocess(args):
""" Process all files """
for data_type in ['tr']:
for data_type in ['tt']:
for speaker in ['mix', 's1', 's2']:
preprocess_one_dir(os.path.join(args.in_dir, data_type, speaker),
os.path.join(args.out_dir, data_type),


Loading…
Cancel
Save