Download Latest Version TextBox 2.0 Release.zip (5.7 MB)
Email in envelope

Get an email when there's a new version of TextBox

Home / v0.2.1
Name Modified Size InfoDownloads / Week
Parent folder
README.md 2021-04-15 828 Bytes
TextBox v0.2.1.tar.gz 2021-04-15 2.3 MB
TextBox v0.2.1.zip 2021-04-15 2.4 MB
Totals: 3 Items   4.6 MB 0

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.
Source: README.md, updated 2021-04-15