BMTrain v0.2.1
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