«ZCls2» is a more faster classification model training framework
After nearly one and a half years of development, ZCls has integrated many training features, includes configuration module, register module, training module, and many model implementations (resnet/mobilenet/senet-sknet-resnest/acbnet-repvgg-dbbnet/ghostnet/gcnet...) and so on. In the development process, it is found that compared with the current excellent classification training framework, such as apex, the training speed of ZCls is not outstanding.
In order to better improve the training speed, we decided to develop a new training framework ZCls2, which is implemented based on apex and provides more friendly and powerful functions. In the preliminary implementation, it can be found that ZCls2 improves the training speed by at least 50% compared with ZCls. More functions are being added.
See Install
See Get started
- zhujian - Initial work - zjykzj
Anyone's participation is welcome! Open an issue or submit PRs.
Small note:
- Git submission specifications should be complied with Conventional Commits
- If versioned, please conform to the Semantic Versioning 2.0.0 specification
- If editing the README, please conform to the standard-readme specification.
Apache License 2.0 © 2022 zjykzj