Skip to content

Releases: alibaba/TorchEasyRec

v0.6.0

30 Oct 03:10
0d34ca6
Compare
Choose a tag to compare

We are excited to announce the release of TorchEasyRec 0.6.0, the first public release for TorchEasyRec.

Major Features and Improvements

  • High-performance training, evaluation, and prediction with GPUs.
  • Supported a variety of input data types, including MaxCompute Table, OSS files, CSV files, Parquet files doc here.
  • Supported a variety of feature types, including IdFeature, RawFeature, ComboFeature, LookupFeature, MatchFeature, ExprFeature, OverlapFeature, TokenizeFeature, SequenceIdFeature, SequenceRawFeature, and SequenceFeature. The feature generation operations is also efficient and robust doc here.
  • Supported a variety of models, including DSSM, TDM, DeepFM, MultiTower, DIN, MMoE, DBMTL, PLE. It is also easy to implement customized models.
  • Supported a variety of loss, including binary_cross_entropy, softmax_cross_entropy, l2_loss, jrc_loss doc here.
  • Supported VariationalDropout feature selection.
  • Easy to deploy a TorchEasyRec model as a high-performance inference service using the TorchEasyRec Processor.

Bug Fixes and Other Changes

  • [bugfix] fix train_eval may hang when use OdpsDataset and set is_orderby_partition=true by @tiankongdeguiji in
  • [bugfix] fix offline predict input tile model with sequence by @tiankongdeguiji in #14

Note

For TorchEasyRec 0.6.x, you should use Docker image version 0.6.

  • For the GPU version (CUDA 12.1):
    • mybigpai-public-registry.cn-beijing.cr.aliyuncs.com/easyrec/tzrec-devel:0.6-cu121
  • For the CPU version:
    • mybigpai-public-registry.cn-beijing.cr.aliyuncs.com/easyrec/tzrec-devel:0.6-cpu

New Contributors

Full Changelog: https://github.com/alibaba/TorchEasyRec/commits/v0.6.0