Deep Java Library (DJL) v0.4.0 Release Notes

Release Date: 2020-03-30 // about 4 years ago
  • ๐Ÿ‘ DJL 0.4.0 brings PyTorch and TensorFlow 2.0 inference support. Now you can use these engines directly from DJL with minimum code changes.

    ๐Ÿš€ Note: TensorFlow 2.0 currently is in PoC stage, users will have to build from source to use it. We expect TF Engine finish in the future releases.

    Key Features

    • Training improvement
      • Add InputStreamTranslator
    • Model Zoo improvement
      • Add LocalZooProvider
      • Add ListModels API
    • ๐Ÿ‘ PyTorch Engine support
      • Use the new ai.djl.pytorch:pytorch-native-auto dependency for automatic engine selection and a simpler build/installation process
      • 60+ methods supported
    • ๐Ÿ‘ PyTorch ModelZoo support
      • Image Classification models: ResNet18 and ResNet50
      • Object Detection model: SSD_ResNet50
    • ๐Ÿ‘ TensorFlow 2.0 Engine support
      • Support on Eager Execution for imperative mode
      • 30+ methods support
    • ๐Ÿ‘ TensorFlow ModelZoo support
      • Image Classification models: ResNet50, MobileNetV2

    ๐Ÿ’ฅ Breaking Changes

    โšก๏ธ There are a few changes in API and ModelZoo packages to adapt to multi-engine support. Please follow our latest examples to update your code base from 0.3.0 to 0.4.0.

    Known Issues

    ๐Ÿš€ 1. PyTorch engine doesn't fully support multithreaded inference. You may see random crashes. Single-threaded inference is not impacted. We expect to fix this issue in a future release. ๐Ÿ’ป 2. We saw random crash on mac for โ€œtransfer Learning on CIFAR-10 Datasetโ€ example on Jupyter Notebook. Command line all works.