Skip to content

v0.6.0.1
Compare
Choose a tag to compare
@thorstenwagner thorstenwagner released this 04 Sep 07:25
· 238 commits to main since this release
v0.6.0.1
effcbe5
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

What's Changed

  • tomotwim_embed.py is now 1.6x faster and linearly scales across multiple GPUs by @thorstenwagner in #40
    • Exploiting new the compile option of the latest pytorch 2.1 nightly build.
    • Internally DistributedDataParallel is used instead of DataParallel
  • Unpin pytorch version by @thorstenwagner in #46

Full Changelog: v0.5.1...v0.6.0.1

Contributors

thorstenwagner
Assets 2
Loading