/diff-bs

Primary LanguagePython

Code for Trainable Set Decoding of Sets of Sequences for Neural Sequence Models

set_decoder contains the main decoder class that implements our approach. decoder implements standard algorithms like sampling, beam search and diverse beam search.

  • Code will be updated SOON!*