Intel® Transfer Learning Tool v0.4.0

New Features

• Support for multi-node training with TensorFlow and Horovod
• Improved SimSiam feature extraction for Anomaly Detection
• Added multi-class datasets to Text Classification Notebooks
• Separate model downloader
• Quantization for TensorFlow Text Classification
• User defined splits for custom image classification models
• Anomaly Detection Workflow
• Vision Workflow
• CutPaste feature extraction for Anomaly Detection
• Added support for PyTorch Hub models
• Added support for Keras Applications
• Improved freeze/unfreeze and list trainable parameters features
• Improved documentation and list of supported models in Models.md

Featured Notebooks:

• Intel® Transfer Learning Tool API tutorial Notebooks:

  • Image Classification Transfer Learning with PyTorch
  • Image Classification Transfer Learning with TensorFlow
  • Text Classification fine-tuning with PyTorch
  • Text Classification fine-tuning with TensorFlow
  • Image Anomaly Detection Transfer Learning with PyTorch

• Intel® Transfer Learning Tool API end-to-end Notebooks:

  • Image Anomaly Detection with PyTorch
  • Document-Level Sentiment Analysis using PyTorch
  • Medical Imaging Classification using TensorFlow
  • Multimodal Cancer Detection using TensorFlow and PyTorch
  • Remote Sensing Image Scene Classification using TensorFlow

• Native Framework Notebooks:

  • Transfer Learning for Image Classification using PyTorch
  • Transfer Learning for Image Classification using TensorFlow
  • Text Classification fine-tuning with PyTorch
  • Text Classification fine-tuning with TensorFlow
  • Question Answering fine-tuning using TensorFlow

Validated configuration

• Ubuntu 20.04/22.04 LTS
• Intel® Optimization for TensorFlow 2.11.0
• Intel® Extension for PyTorch 1.13.0
• PyTorch 1.13.1
• TorchVision 0.14.1
• Intel® Neural Compressor 2.0
• TensorFlow Hub 0.12.0

Known issues

• Multimodal Cancer Detection end-to-end pipeline notebook fails during quantization