Skip to content

v0.2 Release: FP16, Distributed, and Usability updates

Compare
Choose a tag to compare
@raulpuric raulpuric released this 13 Mar 19:41
· 108 commits to master since this release

Our main goal with this release is two-fold:

  • address concerns around usability
  • Update repo with new code for FP16, distributed training

Usability

  • We've brought our training/generation code more in line with the pytorch word language model example
  • Provide PyTorch classifier module/function for classifying sentiment from input text tensor
    • Provide pretrained classifiers/language models for this module
    • Provide simple standalone classifier script/example capable of classifying an input csv/json and writing results to other csv/jsons
  • Flattening our directory structure to make code easier to find
  • Putting reusable PyTorch functionality (new RNN api, weight norm functionality, eventually all fp16 functionality) in its own standalone python module to be published at a later date

FP16 + Distributed