OpenBMB/BMTrain v0.2.3

BMTrain v0.2.3

Repository: OpenBMB/BMTrain

Tag: v0.2.3

Published: 2023-08-17T07:49:23Z

Prerelease: no

Release notes:

What's Changed

• Get rid of torch cpp extension when compiling @MayDomine @Achazwl in #115 #132
• Github action CICD @MayDomine in #115
• Now Loss scale can be managed more dynamic by min and max loss scale by @Achazwl #129
• Fix bmt.load(model) OOM when meets torch >= 1.12 @MayDomine #115
• AdamOffloadOptimizer can choose avx flag automatically in runtime @MayDomine #115
• Now BMTrain is fully compatible with torch 2.0 @MayDomine #115

Full Changelog: https://github.com/OpenBMB/BMTrain/compare/0.2.2...0.2.3