NameError: name 'CriterionType' is not defined #86
Description
I am trying to decode a fine-tuned ASR model,fine-tuned using the vakyansh toolkit using the this script and the following command line:-
python ../../utils/inference/infer.py ${data_path} --task audio_pretraining
--nbest 1 --path ${checkpoint_path} --gen-subset ${subset} --results-path ${result_path} --w2l-decoder viterbi
--lm-weight 2 --word-score -1 --sil-weight 0 --criterion ctc --labels ltr --max-tokens 6000000
--post-process letter --model-overrides "{'w2v_path':'${pretrained_model_path}'}"
But it is showing the following error:-
Traceback (most recent call last):
File "../../utils/inference/infer.py", line 428, in
cli_main()
File "../../utils/inference/infer.py", line 424, in cli_main
main(args)
File "../../utils/inference/infer.py", line 283, in main
generator = build_generator(args)
File "../../utils/inference/infer.py", line 268, in build_generator
return W2lViterbiDecoder(args, task.target_dictionary)
File "/data/Sougata/vakyansh_latest/vakyansh-wav2vec2-experimentation/wav2vec/fairseq/examples/speech_recognition/w2l_decoder.py", line 115, in init
super().init(args, tgt_dict)
File "/data/Sougata/vakyansh_latest/vakyansh-wav2vec2-experimentation/wav2vec/fairseq/examples/speech_recognition/w2l_decoder.py", line 56, in init
self.criterion_type = CriterionType.CTC
NameError: name 'CriterionType' is not defined
Flashlight-text was installed using these set of commands and Flashlight-sequence using these set of commands.
I have also tried installing flashlight-text and flashlight-sequence using the command python -m pip install flashlight-text, but still it is giving the same error.
I have run from flashlight.lib.text.decoder import CriterionType, it is working.