Skip to content

v0.3 Release: Speed & Memory Usage improvements + PyTorch 0.5 updates
Compare
Choose a tag to compare
@raulpuric raulpuric released this 06 Apr 19:41
· 72 commits to master since this release
v0.3
8d1cee1
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.

We've switched our mLSTM model to internally used PyTorch's fused LSTM cell which provides significantly improved GPU memory usage (allowing for larger batch size training) and slight improvements to speed compared to the unfused version we had included in earlier versions.

In order to convert any models you've trained in the past to be usable with this version, please see this issue.

We've also updated our distributed code to address the recent April 3rd changes made to PyTorch's Tensors and Variables.

Assets 2
Loading