hubert icon indicating copy to clipboard operation
hubert copied to clipboard

where are the discrete units weights from?

Open thivux opened this issue 1 year ago • 1 comments

in the train.py file, you have an argument named --warmstart to allow "initialize[ing] from the fairseq HuBERT checkpoint". I wonder which checkpoint is it since fairseq offers a lot of HuBERT checkpoints.

thivux avatar Nov 22 '23 04:11 thivux

I believe it is the HuBERT Base (~95M params) model checkpoint, since that's what they use in the paper.

arjunbahuguna avatar Mar 26 '24 08:03 arjunbahuguna