/cogents

COntrollable GENeration of Text from Slices

Primary LanguagePython

cogents

COntrollable GENeration of Text from Slices

CS236 project

TODO:

  • Debug CUDA assertions thrown midway when training on code dataset
  • Try richer variational prior (e.g., GMM or flow model)