zcls2 0.5.2

More Faster Object Classification Training Framework

Language: 🇺🇸 🇨🇳

«ZCls2» is a more faster classification model training framework

Table of Contents

• Table of Contents
• Background
• Installation
• Usage
• Maintainers
• Thanks
• Contributing
• License

Background

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.

Installation

See Install

Usage

See Get started

Maintainers

• zhujian - Initial work - zjykzj

Thanks

• NVIDIA/apex
• ZJCV/ZCls

Contributing

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.

License

Apache License 2.0 © 2022 zjykzj