Skip to content

OpenNMT-tf 2.5.0

Compare
Choose a tag to compare
@guillaumekln guillaumekln released this 16 Jan 10:44
· 440 commits to master since this release

OpenNMT-tf 2.5.0

New features

  • Update to TensorFlow 2.1
    OpenNMT-tf now depends on the tensorflow pip package instead of tensorflow-gpu. The tensorflow package now includes GPU support by default. If you are upgrading an existing environment, we recommend uninstalling the tensorflow-gpu package before doing so. Also note that TensorFlow 2.1 requires CUDA 10.1.
  • Update to TensorFlow Addons 0.7 with Windows support
  • Data parameter export_vocabulary_assets to control whether vocabularies are exported as file assets or embedded in the graph itself

Fixes and improvements

  • Fix error when reading loss returned by a sequence classifier model
  • Fix error on sequences of length 1 in the sequence tagger with CRF
  • Export vocabularies as file assets by default
  • Remove an unnecessary synchronization when training multiple replicas
  • Internal cleanup to fully remove Python 2 support