Pre-Trained Model Loading Error

[renchrollo]

Hello JusperLee,

The required keys in the state.dict for the model don't match the keys provided when loading the pre-trained model weights (best.pt). Is there a fix?

Ex. "separation.conv1d_list.0.conv1x1.weight" (actual) vs. "separation.conv1d_list.0.0.conv1x1.weight" (expected)

[JusperLee]

maybe you can rename the key for pretrained model