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(