text icon indicating copy to clipboard operation
text copied to clipboard

Published 20 hours ago • verified publisher icon flashlight

Diverse beam search (CPU)

Open joecummings opened this issue 1 year ago • 0 comments

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

joecummings avatar Mar 03 '23 17:03 joecummings