From 4f8e1660affaf84bf753038d508c409d904ae6ef Mon Sep 17 00:00:00 2001 From: huangxu1991 <40886464+huangxu1991@users.noreply.github.com> Date: Fri, 19 Jul 2024 10:33:24 +0800 Subject: [PATCH] Add use_distributed_sampler=False in Trainer (#756) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit if you have defined your own sampler, you should have to set use_distributed_sampler to False! 当使用自定义的 sampler 时,必须设置 use_distributed_sampler 为 False --- GPT_SoVITS/s1_train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/GPT_SoVITS/s1_train.py b/GPT_SoVITS/s1_train.py index ece295d..898ca54 100644 --- a/GPT_SoVITS/s1_train.py +++ b/GPT_SoVITS/s1_train.py @@ -134,6 +134,7 @@ def main(args): logger=logger, num_sanity_val_steps=0, callbacks=[ckpt_callback], + use_distributed_sampler=False, # 非常简单的修改,但解决了采用自定义的 bucket_sampler 下训练步数不一致的问题! ) model: Text2SemanticLightningModule = Text2SemanticLightningModule(