search

NVIDIA / NeMo

A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)
https://docs.nvidia.com/nemo-framework/user-guide/latest/overview.html
Apache License 2.0
11.84k stars 2.46k forks source link

cant load pretrained model #1313

Closed samabdullah closed 3 years ago

samabdullah commented 3 years ago

when i try to upload a pretrained model quartznet = nemo_asr.models.EncDecCTCModel.from_pretrained(model_name="QuartzNet15x5Base-En") it is giving error AttributeError: module 'nemo.collections.asr' has no attribute 'models' [Question]

TottiPuc commented 3 years ago

Could you try to load the state_dict using nemo_asr.models.EncDecCTCModel.from_pretrained(model_name="QuartzNet15x5Base-En", strict=False)? The current work-around is to pass strict=False to either #1297

blisc commented 3 years ago

Can you try from nemo.collections.asr.models import EncDecCTCModel?

okuchaiev commented 3 years ago

please try suggestion from @blisc and re-open if you still see the issue

zhan0502 commented 3 years ago

There is still this error:

AttributeError: type object 'EncDecCTCModel' has no attribute 'restore_from' or 'from_pretrained'

has anyone try out the latest release of nemo 1.3.0?

yipenglinoe commented 3 years ago

There is still this error:

AttributeError: type object 'EncDecCTCModel' has no attribute 'restore_from' or 'from_pretrained'

has anyone try out the latest release of nemo 1.3.0?

i tried , exactly the same issus here "type object 'EncDecCTCModel' has no attribute 'restore_from'" even from_pretrained() can not use too , please check and fix it

blisc commented 3 years ago

Please include a colab notebook in which this issue is reproduced and open another issue.