Unreasonable WER obtained by MMD training in CMatch about easyespnet

[suuuuju]

I made some changes in the code. When decoding, an error occurred that
model.loss.best could not be found.
I did not find the command to save model.loss.best in the code, so I added it to the line after saving snapshot.ep.{epoch}. See the following code for details.
Another change is that when I was doing mmd training, an error occurred: File “/root/model/NeuralSpeech/CMatchASR/utils.py”, line 60, in load_pretrained_model model.load_state_dict(dst_state) RuntimeError: Error(s) in loading state_dict for UDASpeechTransformer: Unexpected key(s) in state_dict: “model”, “optimizer”.
So I changed it to model.load_state_dict(dst_state,strict=False).
What else do I need to check?

save_path = f"{args.outdir}/model.loss.best"  #Specified in the source code
torch_save(model, f"{args.outdir}/snapshot.ep.{epoch}", optimizer=optimizer)
#save model.loss.best
if sum(test_stats['loss_lst'])/len(test_stats['loss_lst']) <= min(test_losses):
    torch_save(model, save_path, optimizer=optimizer)
    early_stop = 0
test_losses.append(sum(test_stats['loss_lst'])/len(test_stats['loss_lst']))