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/