Skip to content

pytorch-optimizer v0.4.0

Choose a tag to compare

View all tags
@kozistr kozistr released this 19 Feb 08:39
v0.4.0
8439f15
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Verified
Learn about vigilant mode.
  • remove .data
  • torch.no_grad() to the all step() function (except closure())
  • support bfloat16 dtype (XLA compatibility)
Assets 2
Loading