openspeech icon indicating copy to clipboard operation
openspeech copied to clipboard

prediction script

Open wuxiuzhi738 opened this issue 4 years ago • 8 comments

❓ Questions & Help

Details

do you have prediction script?

wuxiuzhi738 avatar Sep 06 '21 08:09 wuxiuzhi738

Prediction is not supported yet. We will add more later.

upskyy avatar Sep 08 '21 04:09 upskyy

In hydra_eval.py, line 47, "model = MODEL_REGISTRY[configs.eval.model_name].load_from_checkpoint(configs.eval.checkpoint_path)",
An error occurs that config and tokenizer parameters are missing. I use "model = MODEL_REGISTRY['joint_ctc_conformer_lstm'].load_from_checkpoint(configs=configs, checkpoint_path=configs.eval.checkpoint_path, tokenizer=None)" and do some other modifications, program runs without errors , but the predicted results looks like not right that there are a lot of repetitive words.

wuxiuzhi738 avatar Sep 08 '21 06:09 wuxiuzhi738

As #86 saied.

wuxiuzhi738 avatar Sep 08 '21 06:09 wuxiuzhi738

Can you show me the result?

sooftware avatar Sep 08 '21 06:09 sooftware

image and now ,the train_cer of training is already less than 0.5.

wuxiuzhi738 avatar Sep 08 '21 06:09 wuxiuzhi738

I'm sure it's weird. Can you organize and write both train and evaluation scripts about which model you used?

sooftware avatar Sep 08 '21 06:09 sooftware

model: joint_ctc_conformer_lstm, The training script is basically unchanged Major modifications in hydra_eval.py script: image image

wuxiuzhi738 avatar Sep 08 '21 06:09 wuxiuzhi738

I'm sure it's weird. Can you organize and write both train and evaluation scripts about which model you used?

I wonder if can load the model with 'torch. load',or only use 'load_from_checkpoint' in pytorch-lightninig.

wuxiuzhi738 avatar Sep 09 '21 02:09 wuxiuzhi738