Skip to content

Intel® Transfer Learning Tool v0.6.0

Choose a tag to compare

@ashahba ashahba released this 24 Oct 22:25
· 63 commits to main since this release

New Features

  • Support for LLM Text Generation in TLT API and CLI with custom datasets, including support for 5 new LLM models from HuggingFace
  • New notebooks for PyTorch text generation and HuggingFace image classification
  • Added ability to enable auto mixed precision during evaluation and prediction for image classification as well as TensorFlow text classification

Bug Fixes / Improvements

  • Bug fixes for CLI text classification evaluation command and metrics
  • Improvements to documentation for setup and installation of notebook requirements
  • Bug fix for PyTorch image classification batch not shuffling properly
  • Fixes for the grouped file and dataset columns in the e2e multimodal notebook
  • Fixed batching and accuracy bugs in TensorFlow Multinode and PyTorch Image Classification
  • Fixed minor bug involving Torchvision model weights

New Intel® Transfer Learning Tool API tutorial Notebooks:

  • Text Generation Instruction Tuning LLMs Transfer Learning with PyTorch

New Native Framework Notebooks:

  • Large Language Model Instruction Tuning with PyTorch
  • Image Classification Transfer Learning with HuggingFace

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.15.0

GitHub pages

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