babysor / MockingBird

🚀AI拟声: 5秒内克隆您的声音并生成任意语音内容 Clone a voice in 5 seconds to generate arbitrary speech in real-time
Other
34.88k stars 5.18k forks source link

替换模型之后训练时报错 #890

Open Maplexj opened 1 year ago

Maplexj commented 1 year ago

Summary[问题简述(一句话)] A clear and concise description of what the issue is. 替换模型之后训练时报错 Env & To Reproduce[复现与环境] 描述你用的环境、代码版本、模型 python3.9 mockingbird0.0.1 75k模型 Screenshots[截图(如有)] Arguments: run_id: mandarin syn_dir: F:/AI/mockingbird3/SV2TTS/synthesizer models_dir: synthesizer/saved_models/ save_every: 1000 backup_every: 25000 log_every: 200 force_restart: False hparams:

Checkpoint path: synthesizer\saved_models\mandarin\mandarin.pt Loading training data from: F:\AI\mockingbird3\SV2TTS\synthesizer\train.txt Using model: Tacotron Using device: cuda

Initialising Tacotron Model...

Trainable Parameters: 30.875M

Loading weights at synthesizer\saved_models\mandarin\mandarin.pt Traceback (most recent call last): File "F:\AI\mockingbird3\synthesizer_train.py", line 37, in train(**vars(args)) File "F:\AI\mockingbird3\synthesizer\train.py", line 114, in train model.load(weights_fpath, device, optimizer) File "F:\AI\mockingbird3\synthesizer\models\tacotron.py", line 550, in load optimizer.load_state_dict(checkpoint["optimizer_state"]) File "F:\back\envs\mockingbrid\lib\site-packages\torch\optim\optimizer.py", line 390, in load_state_dict raise ValueError("loaded state dict contains a parameter group " ValueError: loaded state dict contains a parameter group that doesn't match the size of optimizer's group

HaSaKiYasuooo commented 1 year ago

什么替换模型,没看懂