-
Notifications
You must be signed in to change notification settings - Fork 0
Open
Description
When I download the checkpoints from the Google Drive, and run python main_noref.py --filename en-common.csv --modelname self_super.ckpt
I got the following error:
UserWarning: TypedStorage is deprecated. It will be removed in the future and UntypedStorage will be the only storage class. This should only matter to you if you are using storages directly. To access UntypedStorage directly, use tensor.untyped_storage() instead of tensor.storage()
return self.fget.__get__(instance, owner)()
Some weights of XLMRobertaModel were not initialized from the model checkpoint at nreimers/mMiniLMv2-L12-H384-distilled-from-XLMR-Large and are newly initialized: ['roberta.pooler.dense.bias', 'roberta.pooler.dense.weight']
You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference.
Traceback (most recent call last):
File "/share5/users/bruce/asr_projects/asr_metric_research/NoRefER/main_noref.py", line 58, in <module>
model.load_state_dict(model_weights)
File "/share5/users/bruce/miniconda3/envs/tts/lib/python3.9/site-packages/torch/nn/modules/module.py", line 2152, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for aiXER:
Unexpected key(s) in state_dict: "model.embeddings.position_ids".
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels