Name | Modified | Size | Downloads / 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
anddataloader
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.