Browse Source

更新 'train_cload.py'

master
lwj 1 month ago
parent
commit
d53c6a63df
1 changed files with 5 additions and 5 deletions
  1. +5
    -5
      train_cload.py

+ 5
- 5
train_cload.py View File

@@ -59,13 +59,13 @@ parser.add_argument('--device_id', type=int, default=0,
help='Sample rate of audio file')
parser.add_argument('--run_distribute', type=bool, default=True,
help='Sample rate of audio file')
parser.add_argument('--data_batch_size', type=int, default=4,
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/tt',
parser.add_argument('--train', type=str, default='/home/work/user-job-dir/inputs/data_json/tr',
help='Sample rate of audio file')
parser.add_argument('--valid', type=str, default="/home/work/user-job-dir/inputs/data_json/tt",
parser.add_argument('--valid', type=str, default="/home/work/user-job-dir/inputs/data_json/tr",
help='Sample rate of audio file')
parser.add_argument('--test', type=str, default="/home/work/user-job-dir/inputs/data_json/tt",
parser.add_argument('--test', type=str, default="/home/work/user-job-dir/inputs/data_json/tr",
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 ['tt']:
for data_type in ['tr']:
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