Skip to content

Diverse beam search (CPU)#55

Draft
joecummings wants to merge 1 commit intoflashlight:mainfrom
joecummings:diverse-beam-search
Draft

Diverse beam search (CPU)#55
joecummings wants to merge 1 commit intoflashlight:mainfrom
joecummings:diverse-beam-search

Conversation

@joecummings
Copy link

@joecummings joecummings commented Mar 3, 2023

Summary

Adds diverse beam search from this paper.

Helpful Diagrams

Figure 1: Example of how diversity factor (Hamming) is applied at timestep t
Screenshot 2023-03-03 at 1 52 04 PM

Figure 2: General algo
Screenshot 2023-03-03 at 1 51 58 PM

Test Plan (required)

Will add new tests.

Checklist

  • Test coverage
  • Tests pass
  • Code formatted
  • Rebased on latest matter
  • Code documented

Closes #54

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Mar 3, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Diverse beam search

2 participants