Skip to content

TextBox v0.2.1
Compare
Choose a tag to compare
@StevenTang1998 StevenTang1998 released this 15 Apr 13:39
v0.2.1
d04499f
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

TextBox v0.2.1 Release Notes

The TextBox v0.2.1 release includes a number of wonderful new features, some bug fixes and code refactor. A few of the highlights include:

  • We add 6 new models: HRED, CVAE, T5, ProphetNet, Context2Seq and Attribute2Seq.
  • We add 3 new datasets: Persona Chat for dialog system, Amazon Electronic for attribute to text generation and Chinese Classical Poetry Corpus for poem generation.
  • We support Distributed Data Parallel (DDP) for training with multiple GPUs conveniently.
  • We refactor the codes of pretrained language models (PLMs) for improving performance.
  • We refactor the dataset and dataloader to provide unified and convenient interface.
  • We unify and simplify the generate function for each model.
  • We unify the config parameters of different models and datasets.
Assets 2
Loading