Export Finetuned wav2vec2 model

[thevasudevgupta]

Any pull request you open is subject to the TensorFlow Hub Terms of Service at www.tfhub.dev/terms and Google's Privacy Policy at https://www.google.com/policies/privacy.

This PR adds fine-tuned wav2vec2 model to TFHub. This model can directly be used for inference. It achieved ~ 6% WER on LibriSpeech test-clean dataset.

@MorganR @sayakpaul

[sayakpaul]

@vasudevgupta7 please work on the failure logs too.

[thevasudevgupta]

@vasudevgupta7 please work on the failure logs too.

I am actually unable to understand the reason for failing logs. Any help is much appreciated.

[sayakpaul]

I am actually unable to understand the reason for failing logs. Any help is much appreciated.

Strange that it's failing due to {'vasudevgupta7/models/wav2vec2-960h/1.md': 'Could not read tarfile.'}. You seem to have provided the correct tar path. Could it be that the directory structure is not correct as per the format the Hub expects?

In any case, @MorganR will be able to tell more.

[MorganR]

I am actually unable to understand the reason for failing logs. Any help is much appreciated.

Strange that it's failing due to {'vasudevgupta7/models/wav2vec2-960h/1.md': 'Could not read tarfile.'}. You seem to have provided the correct tar path. Could it be that the directory structure is not correct as per the format the Hub expects?

In any case, @MorganR will be able to tell more.

I would expect this to work too. I can investigate next week.

[thevasudevgupta]

I would expect this to work too. I can investigate next week.

Thanks!!

Rest all suggestions are done.

[MorganR]

Hey Vasudev, it looks like the model asset is an archive, but it is not compressed. You can compress the model like this:

$ cd $SAVED_MODEL_DIR
$ tar -czvf ../saved-model.tar.gz *

That should fix the validation issue.

[thevasudevgupta]

@MorganR, I have changed the saved-model.tar-gz file in my GCS bucket. Weights link is still the same as before. Can you re-run the tests?

[WGierke]

Hi @vasudevgupta7
It looks like the archive contains some files that were auto-generated by OSX:

$ tar -xvf saved-model.tar.gz 
._assets
assets/
._saved_model.pb
saved_model.pb
._variables
variables/
variables/._variables.data-00000-of-00001
variables/variables.data-00000-of-00001
variables/._variables.index
variables/variables.index
$ cat variables/._variables.data-00000-of-00001 
Mac OS X        	2��ATTR��<�<com.apple.quarantineq/0083;611b559f;Safari...

Would you mind removing these files starting with ._, please?
Note that you can also execute the validation script yourself locally by running

python tools/validator.py vasudevgupta7/models/wav2vec2-960h/1.md

Thank you!

[thevasudevgupta]

@WGierke, I have deleted extra files & tests passing for me locally.

[github-actions]

Thank you for your contribution. Your pull request has been accepted according to the TensorFlow Hub Terms of Service at www.tfhub.dev/terms and Google's Privacy Policy at https://www.google.com/policies/privacy. Your model should appear on tfhub.dev within a day.