Skip to content

Intel® Transfer Learning Tool v0.5.0

Compare
Choose a tag to compare
@ashahba ashahba released this 24 Jun 01:14
· 153 commits to main since this release

New Features

  • Support for Hugging Face text classification models using TensorFlow
  • Support for Keras model preprocessors
  • Disributed training Dockerfiles and Helm chart
  • Optimized CPU settings for TensorFlow and PyTorch
  • Intel® Neural Compressor 2.1.1 API migrations
  • Virtualenv support for multinode training with PyTorch
  • Added overwrite option for quantization and optimize_graph APIs
  • Improved AI Kit compatibility
  • Updated vision workflow
  • Improved design and usability of documentation

New Notebooks:

  • Two performance comparison notebooks for image classification and text classification

Bug fixes:

  • multi-class text classification with TensorFlow

Validated configuration

  • Ubuntu 20.04/22.04 LTS
  • Intel® Optimization for TensorFlow 2.12.0
  • Intel® Extension for PyTorch 1.13.100
  • PyTorch 1.13.1
  • TorchVision 0.14.1
  • Intel® Neural Compressor 2.1.1
  • TensorFlow Hub 0.13.0

Known issues

  • The CLI's benchmarking command fails for some models, such as TensorFlow text classification. Recommend using the API for advanced workflows.

GitHub pages

https://intelai.github.io/transfer-learning/v0.5.0/