tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

change custom sampler to default

Browse Source
This commit is contained in:
litagin02
2024-05-29 05:46:31 +09:00
parent ec12381acf
commit f4f94d4c20
4 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
train_ms_jp_extra.py
View File

@@ -99,8 +99,8 @@ def run():
default=None,
)
parser.add_argument(
"--use_custom_batch_sampler",
help="Use custom batch sampler for training, which was used in the version < 2.5",
"--not_use_custom_batch_sampler",
help="Don't use custom batch sampler for training, which was used in the version < 2.5",
action="store_true",
)
args = parser.parse_args()
@@ -219,7 +219,7 @@ def run():
writer_eval = SummaryWriter(log_dir=os.path.join(model_dir, "eval"))
train_dataset = TextAudioSpeakerLoader(hps.data.training_files, hps.data)
collate_fn = TextAudioSpeakerCollate(use_jp_extra=True)
if args.use_custom_batch_sampler:
if not args.not_use_custom_batch_sampler:
train_sampler = DistributedBucketSampler(
train_dataset,
hps.train.batch_size,