Skip to content

pytorch-optimizer v2.1.1

Compare
Choose a tag to compare
@kozistr kozistr released this 02 Jan 12:18
503ad2e

Change Log

#90

Feature

  • Support gradient centralization for Adai optimizer
  • Support AdamD debias for AdaPNM optimizer
  • Register custom exceptions (e.g. NoSparseGradientError, NoClosureError, ...)

Documentation

  • Add API documentation

Bug

  • Fix SAM optimizer