Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Beam search #30

Open
vinevix opened this issue Mar 20, 2023 · 7 comments
Open

Beam search #30

vinevix opened this issue Mar 20, 2023 · 7 comments

Comments

@vinevix
Copy link

vinevix commented Mar 20, 2023

Hi, is there a way to apply beam search in order to generate text?

@buxiangzhiren
Copy link
Owner

buxiangzhiren commented Mar 20, 2023

We don't find a way to apply beam search. We tried to change the argmax to sample for achieving beam search, but the performance dropped.

@vinevix
Copy link
Author

vinevix commented Mar 20, 2023 via email

@buxiangzhiren
Copy link
Owner

The difference is that “generate2_adpt_if”uses the image-free strategy in inference. Maybe you can change the variable "scale" to get different results. But I am not sure that the differences of results from various "scale" are great to support the reinforcement learning.

@vinevix
Copy link
Author

vinevix commented Mar 20, 2023 via email

@buxiangzhiren
Copy link
Owner

The range was reported in Figure 4 of our paper. For various results, I suggest that you set the range 1~4.
image

@buxiangzhiren
Copy link
Owner

The range was reported in Figure 4 of our paper. For various results, I suggest that you set the range 1~4. image

The results are reported in validation dataset. When "scale" is 1.06, our model can achieve the best performance in test dataset.

@vinevix
Copy link
Author

vinevix commented Mar 20, 2023 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants