An error occurs during processing
Start from checkpoint: C:\Users\Kveris\Desktop\Music Source Separation Training 1.0.12\checkpoint\bs_exp_siameseroformer.ckpt
Traceback (most recent call last):
File "C:\Users\Kveris\Desktop\Music Source Separation Training 1.0.12\inference.py", line 184, in
proc_folder(None)
File "C:\Users\Kveris\Desktop\Music Source Separation Training 1.0.12\inference.py", line 169, in proc_folder
model.load_state_dict(state_dict)
File "C:\Users\Kveris\AppData\Local\Programs\Python\Python311\Lib\site-packages\torch\nn\modules\module.py", line 2635, in load_state_dict
raise RuntimeError(
RuntimeError: Error(s) in loading state_dict for BSRoformer:
Unexpected key(s) in state_dict: "layers.0.0.ln_y_attn.0.gamma", "layers.0.0.ln_x_attn.0.gamma", "layers.0.0.ln_y_mlp.0.gamma", "layers.0.0.ln_x_mlp.0.gamma", "layers.0.0.y_attn.0", "layers.0.1.ln_y_attn.0.gamma", "layers.0.1.ln_x_attn.0.gamma", "layers.0.1.ln_y_mlp.0.gamma", "layers.0.1.ln_x_mlp.0.gamma", "layers.0.1.y_attn.0", "layers.1.0.ln_y_attn.0.gamma", "layers.1.0.ln_x_attn.0.gamma", "layers.1.0.ln_y_mlp.0.gamma", "layers.1.0.ln_x_mlp.0.gamma", "layers.1.0.y_attn.0", "layers.1.1.ln_y_attn.0.gamma", "layers.1.1.ln_x_attn.0.gamma", "layers.1.1.ln_y_mlp.0.gamma", "layers.1.1.ln_x_mlp.0.gamma", "layers.1.1.y_attn.0", "layers.2.0.ln_y_attn.0.gamma", "layers.2.0.ln_x_attn.0.gamma", "layers.2.0.ln_y_mlp.0.gamma", "layers.2.0.ln_x_mlp.0.gamma", "layers.2.0.y_attn.0", "layers.2.1.ln_y_attn.0.gamma", "layers.2.1.ln_x_attn.0.gamma", "layers.2.1.ln_y_mlp.0.gamma", "layers.2.1.ln_x_mlp.0.gamma", "layers.2.1.y_attn.0", "layers.3.0.ln_y_attn.0.gamma", "layers.3.0.ln_x_attn.0.gamma", "layers.3.0.ln_y_mlp.0.gamma", "layers.3.0.ln_x_mlp.0.gamma", "layers.3.0.y_attn.0", "layers.3.1.ln_y_attn.0.gamma", "layers.3.1.ln_x_attn.0.gamma", "layers.3.1.ln_y_mlp.0.gamma", "layers.3.1.ln_x_mlp.0.gamma", "layers.3.1.y_attn.0", "layers.4.0.ln_y_attn.0.gamma", "layers.4.0.ln_x_attn.0.gamma", "layers.4.0.ln_y_mlp.0.gamma", "layers.4.0.ln_x_mlp.0.gamma", "layers.4.0.y_attn.0", "layers.4.1.ln_y_attn.0.gamma", "layers.4.1.ln_x_attn.0.gamma", "layers.4.1.ln_y_mlp.0.gamma", "layers.4.1.ln_x_mlp.0.gamma", "layers.4.1.y_attn.0", "layers.5.0.ln_y_attn.0.gamma", "layers.5.0.ln_x_attn.0.gamma", "layers.5.0.ln_y_mlp.0.gamma", "layers.5.0.ln_x_mlp.0.gamma", "layers.5.0.y_attn.0", "layers.5.1.ln_y_attn.0.gamma", "layers.5.1.ln_x_attn.0.gamma", "layers.5.1.ln_y_mlp.0.gamma", "layers.5.1.ln_x_mlp.0.gamma", "layers.5.1.y_attn.0", "layers.6.0.ln_y_attn.0.gamma", "layers.6.0.ln_x_attn.0.gamma", "layers.6.0.ln_y_mlp.0.gamma", "layers.6.0.ln_x_mlp.0.gamma", "layers.6.0.y_attn.0", "layers.6.1.ln_y_attn.0.gamma", "layers.6.1.ln_x_attn.0.gamma", "layers.6.1.ln_y_mlp.0.gamma", "layers.6.1.ln_x_mlp.0.gamma", "layers.6.1.y_attn.0", "layers.7.0.ln_y_attn.0.gamma", "layers.7.0.ln_x_attn.0.gamma", "layers.7.0.ln_y_mlp.0.gamma", "layers.7.0.ln_x_mlp.0.gamma", "layers.7.0.y_attn.0", "layers.7.1.ln_y_attn.0.gamma", "layers.7.1.ln_x_attn.0.gamma", "layers.7.1.ln_y_mlp.0.gamma", "layers.7.1.ln_x_mlp.0.gamma", "layers.7.1.y_attn.0", "layers.8.0.ln_y_attn.0.gamma", "layers.8.0.ln_x_attn.0.gamma", "layers.8.0.ln_y_mlp.0.gamma", "layers.8.0.ln_x_mlp.0.gamma", "layers.8.0.y_attn.0", "layers.8.1.ln_y_attn.0.gamma", "layers.8.1.ln_x_attn.0.gamma", "layers.8.1.ln_y_mlp.0.gamma", "layers.8.1.ln_x_mlp.0.gamma", "layers.8.1.y_attn.0", "layers.9.0.ln_y_attn.0.gamma", "layers.9.0.ln_x_attn.0.gamma", "layers.9.0.ln_y_mlp.0.gamma", "layers.9.0.ln_x_mlp.0.gamma", "layers.9.0.y_attn.0", "layers.9.1.ln_y_attn.0.gamma", "layers.9.1.ln_x_attn.0.gamma", "layers.9.1.ln_y_mlp.0.gamma", "layers.9.1.ln_x_mlp.0.gamma", "layers.9.1.y_attn.0", "layers.10.0.ln_y_attn.0.gamma", "layers.10.0.ln_x_attn.0.gamma", "layers.10.0.ln_y_mlp.0.gamma", "layers.10.0.ln_x_mlp.0.gamma", "layers.10.0.y_attn.0", "layers.10.1.ln_y_attn.0.gamma", "layers.10.1.ln_x_attn.0.gamma", "layers.10.1.ln_y_mlp.0.gamma", "layers.10.1.ln_x_mlp.0.gamma", "layers.10.1.y_attn.0", "layers.11.0.ln_y_attn.0.gamma", "layers.11.0.ln_x_attn.0.gamma", "layers.11.0.ln_y_mlp.0.gamma", "layers.11.0.ln_x_mlp.0.gamma", "layers.11.0.y_attn.0", "layers.11.1.ln_y_attn.0.gamma", "layers.11.1.ln_x_attn.0.gamma", "layers.11.1.ln_y_mlp.0.gamma", "layers.11.1.ln_x_mlp.0.gamma", "layers.11.1.y_attn.0".
[0mProcess failed with return code 1
What should I do in this situation? I'm trying to run the model through MSST.
You need to replace the one in models/bs_roformer with the bs_roformer.py in this repository