search

facebookresearch / textlesslib

Library for Textless Spoken Language Processing
MIT License
528 stars 50 forks source link

Load mhubert_base_25hz error #33

Closed 0nutation closed 7 months ago

0nutation commented 7 months ago

Hi, when I load mhubert 25hz, I encounter error like: 

RuntimeError: Error(s) in loading state_dict for HubertModel:
        Missing key(s) in state_dict: "encoder.pos_conv.0.weight_g", "encoder.pos_conv.0.weight_v". 
        Unexpected key(s) in state_dict: "encoder.pos_conv.0.weight", "encoder.pos_conv.0.running_mean", "encoder.pos_conv.0.running_var", "encoder.pos_conv.0.num_batches_tracked", "encoder.pos_conv.1.weight", "encoder.pos_conv.1.bias".

It seems to origin from different version of pos_conv implentation, how can I fix it?