Skip to content

BMTrain v0.2.1

Compare
Choose a tag to compare
@a710128 a710128 released this 30 Jan 03:18
· 134 commits to main since this release

What's Changed

  • fix output shape mismatch after CheckpointBlock by @Achazwl in #64
  • add test for grad accumulation and state_dict interface. by @MayDomine in #61
  • fix inspect grad mean/std from None to 0 by @Achazwl in #60

Full Changelog: 0.2.0...0.2.1