Skip to content

Releases: kozistr/pytorch_optimizer

pytorch-optimizer v0.6.0

02 Apr 14:27
6ba157d

Choose a tag to compare

feature: pnm optimizer

pytorch-optimizer v0.5.0

05 Mar 18:48
f66a612

Choose a tag to compare

feature: shampoo optimizer

pytorch-optimizer v0.4.2

05 Mar 07:36
d657060

Choose a tag to compare

refactor: coverage

pytorch-optimizer v0.4.1

20 Feb 09:29
6db0d49

Choose a tag to compare

refactor: coverage

pytorch-optimizer v0.4.0

19 Feb 08:39
8439f15

Choose a tag to compare

  • remove .data
  • torch.no_grad() to the all step() function (except closure())
  • support bfloat16 dtype (XLA compatibility)

pytorch-optimizer v0.3.7

01 Feb 05:07
3cd5158

Choose a tag to compare

feature: LARS optimizer

pytorch-optimizer v0.3.6

31 Jan 06:44
16aeb2c

Choose a tag to compare

refactor: utils

pytorch-optimizer v0.3.5

30 Jan 07:53
62f518b

Choose a tag to compare

  • Refactor test modules
    • merge fp32 & fp16 recipes into one list
  • test cases for Lookahead
  • test case for no gradient
  • fix Ranger21 optimizer able to handle no gradient
  • improve stability pre_norm for Lamb optimizer

pytorch-optimizer v0.3.4

29 Jan 12:15
5538b6b

Choose a tag to compare

fix: MADGRAD optimizer

pytorch-optimizer v0.3.3

29 Jan 07:53
56c05b9

Choose a tag to compare

feature: RaLamb optimizer