MMEngine Release V0.7.0
v0.7.0 (03/16/2023)
Highlights
- Support PyTorch 2.0! Accelerate training by compiling models. See the tutorial Model Compilation for details
- Add
EarlyStoppingHook
to stop training when the metric does not improve
New Features & Enhancements
- Add configurations to support
torch.compile
in Runner by @C1rN09 in #976 - Support
EarlyStoppingHook
by @nijkah in #739 - Disable duplicated warning during distributed training by @HAOCHENYE in #961
- Add
FUNCTIONS
root Registry by @HAOCHENYE in #983 - Save the "memory" field to visualization backends by @enkilee in #974
- Enable bf16 in
AmpOptimWrapper
by @C1rN09 in #960 - Support writing data to
vis_backend
with prefix by @HAOCHENYE in #972 - Support exporting logs of different ranks in debug mode by @HAOCHENYE in #968
- Silence error when
ManagerMixin
built instance with duplicate name. by @HAOCHENYE in #990
Bug fixes
- Fix optim_wrapper unittest for
pytorch < 1.10.0
by @C1rN09 in #975 - Support calculating the flops of
matmul
with single dimension matrix by @HAOCHENYE in #970 - Fix repeated warning by @HAOCHENYE in #992
- Fix lint by @zhouzaida in #993
- Fix AMP in Ascend and support using NPUJITCompile environment by @luomaoling in #994
- Fix inferencer gets wrong configs path by @HAOCHENYE in #996
Docs
- Translate "Debug Tricks" to English by @enkilee in #953
- Translate "Model Analysis" document to English by @enkilee in #956
- Translate "Model Complexity Analysis" to Chinese. by @VoyagerXvoyagerx in #969
- Add a document about setting interval by @YuetianW in #964
- Translate "how to set random seed" by @xin-li-67 in #930
- Fix typo by @zhouzaida in #965
- Fix typo in hook document by @acdart in #980
- Fix changelog date by @HAOCHENYE in #986
New Contributors
- @YuetianW made their first contribution in #964
- @enkilee made their first contribution in #953
- @acdart made their first contribution in #980
- @VoyagerXvoyagerx made their first contribution in #969
Full Changelog: v0.6.0...v0.7.0