search

NeuSpeech / EEG-To-Text

72 stars 8 forks source link

Pretrained weights for Step 1 not found #5

Closed Dark-knight-02 closed 2 weeks ago

Dark-knight-02 commented 1 month ago

I was trying to reproduce the paper and the pretrained weights for step 1 'path_to_step_1_checkpoint.pt' was not in the GitHub. Could you please upload the weights?

NeuSpeech commented 4 weeks ago

I think you don't need pretrained weights. It only trains the step2

Dark-knight-02 commented 4 weeks ago

Yes while I was trying to regenerate the paper on default settings the BeLu-1 score that I am getting is around 0.24 and not that mentioned in the results of the paper. Is there any suggestions to improve the score or on the exact configuration to receive the best result?

Dark-knight-02 commented 4 weeks ago

If possible can you please share the path_to_step_1_checkpoint.pt file?

NeuSpeech commented 3 weeks ago

there is no step_1_checkpoint.pt file. we updated code in main branch to reproduce the results in paper. please check out.