Error when export nemo2riva
Does the latest model support convert nemo2riva
I got error when export using nemo2riva 2.22.0
Traceback (most recent call last):
File "/usr/local/bin/nemo2riva", line 8, in
sys.exit(nemo2riva())
File "/usr/local/lib/python3.10/site-packages/nemo2riva/cli/nemo2riva.py", line 49, in nemo2riva
Nemo2Riva(args)
File "/usr/local/lib/python3.10/site-packages/nemo2riva/convert.py", line 59, in Nemo2Riva
raise e
File "/usr/local/lib/python3.10/site-packages/nemo2riva/convert.py", line 52, in Nemo2Riva
model = ModelPT.restore_from(restore_path=nemo_in, trainer=trainer)
File "/usr/local/lib/python3.10/site-packages/nemo/core/classes/modelPT.py", line 501, in restore_from
instance = cls._save_restore_connector.restore_from(
File "/usr/local/lib/python3.10/site-packages/nemo/core/connectors/save_restore_connector.py", line 284, in restore_from
self.load_instance_with_state_dict(instance, state_dict, strict)
File "/usr/local/lib/python3.10/site-packages/nemo/core/connectors/save_restore_connector.py", line 231, in load_instance_with_state_dict
instance.load_state_dict(state_dict, strict=strict)
File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 2629, in load_state_dict
raise RuntimeError(
RuntimeError: Error(s) in loading state_dict for EncDecFrameClassificationModel:
Missing key(s) in state_dict: "loss.weight".