Hi, thanks for uploading the project. I have the following issue.
I downloaded the src and the tar file and changed the path in the toml. This is the command I run.
python inference.py -C config/inference/fullsubnet.toml -M /content/fullsubnet_best_model_58epochs.tar -O /content/output
Using CPU in the experiment.
Loading inference dataset...
Num of noisy files in /content/input: 1
Loading model...
当前正在处理 tar 格式的模型断点,其 epoch 为:58.
Traceback (most recent call last):
File "inference.py", line 37, in <module>
main(configuration, checkpoint_path, output_dir)
File "inference.py", line 15, in main
output_dir
File "/content/FullSubNet-0.1/src/inferencer/DNS_INTERSPEECH.py", line 52, in __init__
super().__init__(config, checkpoint_path, output_dir)
File "/content/FullSubNet-0.1/src/common/inferencer.py", line 25, in __init__
self.model, epoch = self._load_model(config["model"], checkpoint_path, self.device)
File "/content/FullSubNet-0.1/src/common/inferencer.py", line 94, in _load_model
model.load_state_dict(new_state_dict)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 1407, in load_state_dict
self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for Model:
Missing key(s) in state_dict: "fband_model.sequence_model.weight_ih_l0", "fband_model.sequence_model.weight_hh_l0", "fband_model.sequence_model.bias_ih_l0", "fband_model.sequence_model.bias_hh_l0", "fband_model.sequence_model.weight_ih_l1", "fband_model.sequence_model.weight_hh_l1", "fband_model.sequence_model.bias_ih_l1", "fband_model.sequence_model.bias_hh_l1", "fband_model.fc_output_layer.weight", "fband_model.fc_output_layer.bias", "sband_model.sequence_model.weight_ih_l0", "sband_model.sequence_model.weight_hh_l0", "sband_model.sequence_model.bias_ih_l0", "sband_model.sequence_model.bias_hh_l0", "sband_model.sequence_model.weight_ih_l1", "sband_model.sequence_model.weight_hh_l1", "sband_model.sequence_model.bias_ih_l1", "sband_model.sequence_model.bias_hh_l1", "sband_model.fc_output_layer.weight", "sband_model.fc_output_layer.bias".
Unexpected key(s) in state_dict: "l.sequence_model.weight_ih_l0", "l.sequence_model.weight_hh_l0", "l.sequence_model.bias_ih_l0", "l.sequence_model.bias_hh_l0", "l.sequence_model.weight_ih_l1", "l.sequence_model.weight_hh_l1", "l.sequence_model.bias_ih_l1", "l.sequence_model.bias_hh_l1", "l.fc_output_layer.weight", "l.fc_output_layer.bias". ```